Files
swift-mirror/test/AutoDiff/SILOptimizer/Inputs/differentiation_diagnostics_other_file.swift
Richard Wei af8942d940 [AutoDiff] Rename '@differentiable' to '@differentiable(reverse)'.
Compiler:
- Add `Forward` and `Reverse` to `DifferentiabilityKind`.
- Expand `DifferentiabilityMask` in `ExtInfo` to 3 bits so that it now holds all 4 cases of `DifferentiabilityKind`.
- Parse `@differentiable(reverse)` and `@differentiable(_forward)` declaration attributes and type attributes.
- Emit a warning for `@differentiable` without `reverse`.
- Emit an error for `@differentiable(_forward)`.
- Rename `@differentiable(linear)` to `@differentiable(_linear)`.
- Make `@differentiable(reverse)` type lowering go through today's `@differentiable` code path. We will specialize it to reverse-mode in a follow-up patch.

ABI:
- Add `Forward` and `Reverse` to `FunctionMetadataDifferentiabilityKind`.
- Extend `TargetFunctionTypeFlags` by 1 bit to store the highest bit of differentiability kind (linear). Note that there is a 2-bit gap in `DifferentiabilityMask` which is reserved for `AsyncMask` and `ConcurrentMask`; `AsyncMask` is ABI-stable so we cannot change that.

_Differentiation module:
- Replace all occurrences of `@differentiable` with `@differentiable(reverse)`.
- Delete `_transpose(of:)`.

Resolves rdar://69980056.
2021-02-07 14:09:46 -08:00

55 lines
1.2 KiB
Swift

import _Differentiation
protocol Protocol: Differentiable {
// Test cross-file `@differentiable` attribute.
@differentiable(reverse, wrt: self)
func identityDifferentiableAttr() -> Self
// Test `@differentiable` propagation from storage declaration to accessors.
@differentiable(reverse)
var property: Float { get set }
// Test `@differentiable` propagation from storage declaration to accessors.
@differentiable(reverse)
subscript() -> Float { get set }
}
extension Protocol {
func identityDerivativeAttr() -> Self { self }
// Test cross-file `@derivative` attribute.
@derivative(of: identityDerivativeAttr)
func vjpIdentityDerivativeAttr() -> (
value: Self, pullback: (TangentVector) -> TangentVector
) {
fatalError()
}
}
class Class: Differentiable {
// Test `@differentiable` propagation from storage declaration to accessors.
@differentiable(reverse)
var property: Float {
get { 1 }
set {}
}
// Test `@differentiable` propagation from storage declaration to accessors.
@differentiable(reverse)
subscript() -> Float {
get { 1 }
set {}
}
}
struct S: Differentiable {
var value: Float
}
extension Array where Element == S {
@differentiable(reverse)
func sum() -> Float {
return 0
}
}