mirror of
https://github.com/apple/swift.git
synced 2025-12-21 12:14:44 +01:00
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.
55 lines
1.2 KiB
Swift
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
|
|
}
|
|
}
|