mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
af8942d940
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.
78 lines
3.3 KiB
Swift
78 lines
3.3 KiB
Swift
// RUN: %target-swift-frontend -emit-silgen -verify %s | %FileCheck %s
|
|
// REQUIRES: asserts
|
|
|
|
import _Differentiation
|
|
|
|
@noDerivative var flag: Bool
|
|
|
|
struct NotDifferentiable {
|
|
@noDerivative var stored: Float
|
|
|
|
@noDerivative
|
|
var computedProperty: Float {
|
|
get { 1 }
|
|
set {}
|
|
_modify { yield &stored }
|
|
}
|
|
|
|
@noDerivative
|
|
func instanceMethod(_ x: Float) -> Float { x }
|
|
|
|
@noDerivative
|
|
static func staticMethod(_ x: Float) -> Float { x }
|
|
|
|
@noDerivative
|
|
subscript(_ x: Float) -> Float {
|
|
get { 1 }
|
|
set {}
|
|
_modify { yield &stored }
|
|
}
|
|
}
|
|
|
|
// CHECK-LABEL: struct NotDifferentiable {
|
|
// CHECK: @noDerivative @_hasStorage var stored: Float { get set }
|
|
// CHECK: @noDerivative var computedProperty: Float { get set _modify }
|
|
// CHECK: @noDerivative func instanceMethod(_ x: Float) -> Float
|
|
// CHECK: @noDerivative static func staticMethod(_ x: Float) -> Float
|
|
// CHECK: @noDerivative subscript(x: Float) -> Float { get set _modify }
|
|
// CHECK: }
|
|
|
|
// CHECK-LABEL: // NotDifferentiable.computedProperty.getter
|
|
// CHECK: sil hidden [_semantics "autodiff.nonvarying"] [ossa] @$s17noderivative_attr17NotDifferentiableV16computedPropertySfvg : $@convention(method) (NotDifferentiable) -> Float
|
|
|
|
// CHECK-LABEL: // NotDifferentiable.computedProperty.setter
|
|
// CHECK: sil hidden [_semantics "autodiff.nonvarying"] [ossa] @$s17noderivative_attr17NotDifferentiableV16computedPropertySfvs : $@convention(method) (Float, @inout NotDifferentiable) -> ()
|
|
|
|
// CHECK-LABEL: // NotDifferentiable.computedProperty.modify
|
|
// CHECK: sil hidden [_semantics "autodiff.nonvarying"] [ossa] @$s17noderivative_attr17NotDifferentiableV16computedPropertySfvM : $@yield_once @convention(method) (@inout NotDifferentiable) -> @yields @inout Float
|
|
|
|
// CHECK-LABEL: // NotDifferentiable.instanceMethod(_:)
|
|
// CHECK: sil hidden [_semantics "autodiff.nonvarying"] [ossa] @$s17noderivative_attr17NotDifferentiableV14instanceMethodyS2fF : $@convention(method) (Float, NotDifferentiable) -> Float
|
|
|
|
// CHECK-LABEL: // static NotDifferentiable.staticMethod(_:)
|
|
// CHECK: sil hidden [_semantics "autodiff.nonvarying"] [ossa] @$s17noderivative_attr17NotDifferentiableV12staticMethodyS2fFZ : $@convention(method) (Float, @thin NotDifferentiable.Type) -> Float
|
|
|
|
// CHECK-LABEL: // NotDifferentiable.subscript.getter
|
|
// CHECK: sil hidden [_semantics "autodiff.nonvarying"] [ossa] @$s17noderivative_attr17NotDifferentiableVyS2fcig : $@convention(method) (Float, NotDifferentiable) -> Float
|
|
|
|
// CHECK-LABEL: // NotDifferentiable.subscript.setter
|
|
// CHECK: sil hidden [_semantics "autodiff.nonvarying"] [ossa] @$s17noderivative_attr17NotDifferentiableVyS2fcis : $@convention(method) (Float, Float, @inout NotDifferentiable) -> ()
|
|
|
|
// CHECK-LABEL: // NotDifferentiable.subscript.modify
|
|
// CHECK: sil hidden [_semantics "autodiff.nonvarying"] [ossa] @$s17noderivative_attr17NotDifferentiableVyS2fciM : $@yield_once @convention(method) (Float, @inout NotDifferentiable) -> @yields @inout Float
|
|
|
|
struct Bar: Differentiable {
|
|
@noDerivative var stored: Float
|
|
}
|
|
|
|
// Test TF-152: derived conformances "no interface type set" crasher.
|
|
struct TF_152: Differentiable {
|
|
@differentiable(reverse, wrt: bar)
|
|
func applied(to input: Float, bar: TF_152_Bar) -> Float {
|
|
return input
|
|
}
|
|
}
|
|
struct TF_152_Bar: Differentiable {
|
|
@noDerivative let dense: Float
|
|
}
|