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.
35 lines
860 B
Swift
35 lines
860 B
Swift
// RUN: %empty-directory(%t)
|
|
// RUN: %target-build-swift-dylib(%t/%target-library-name(Library)) -emit-module -emit-module-path %t/Library.swiftmodule -module-name Library -DLIBRARY %s
|
|
// RUN: %target-build-swift -I %t -O -emit-module %s
|
|
|
|
// SR-14004: Assertion failure due to function with `differentiable_function_extract`
|
|
// with explicit extractee type being deserialized into a raw SIL module.
|
|
|
|
#if LIBRARY
|
|
|
|
import _Differentiation
|
|
|
|
public struct Struct<Scalar>: Differentiable {}
|
|
|
|
@differentiable(reverse)
|
|
public func foo<Scalar>(_ x: Struct<Scalar>) -> Struct<Scalar> { x }
|
|
|
|
@inlinable
|
|
@differentiable(reverse)
|
|
public func bar<Scalar>(_ x: Struct<Scalar>) -> Struct<Scalar> {
|
|
foo(x)
|
|
}
|
|
|
|
#else
|
|
|
|
import _Differentiation
|
|
import Library
|
|
|
|
public func foo(
|
|
body: @differentiable(reverse) (Struct<Float>) -> Struct<Float> = bar
|
|
) {
|
|
fatalError()
|
|
}
|
|
|
|
#endif
|