mirror of
https://github.com/apple/swift.git
synced 2025-12-21 12:14:44 +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.
67 lines
2.4 KiB
Swift
67 lines
2.4 KiB
Swift
// RUN: %target-swift-frontend -emit-sil -verify -primary-file %s %S/Inputs/differentiation_diagnostics_other_file.swift -module-name main -o /dev/null
|
|
|
|
// Test differentiation transform cross-file diagnostics.
|
|
|
|
import _Differentiation
|
|
|
|
// TF-1271: Test `@differentiable` original function in other file.
|
|
@differentiable(reverse)
|
|
func crossFileDifferentiableAttr<T: Protocol>(
|
|
_ input: T
|
|
) -> T {
|
|
return input.identityDifferentiableAttr()
|
|
}
|
|
|
|
// TF-1272: Test original function with registered derivatives in other files.
|
|
// FIXME(TF-1272): Find a way to type-check `@derivative` attributes in other
|
|
// files.
|
|
@differentiable(reverse)
|
|
func crossFileDerivativeAttr<T: Protocol>(
|
|
_ input: T
|
|
) -> T {
|
|
// expected-error @+2 {{expression is not differentiable}}
|
|
// expected-note @+1 {{cannot differentiate functions that have not been marked '@differentiable' and that are defined in other files}}
|
|
return input.identityDerivativeAttr()
|
|
}
|
|
|
|
// TF-1234: Test `@differentiable` propagation from protocol requirement storage
|
|
// declarations to their accessors in other file.
|
|
|
|
@differentiable(reverse)
|
|
func protocolRequirementGetters<T: Protocol>(_ x: T) -> Float {
|
|
x.property + x[]
|
|
}
|
|
|
|
// TODO(TF-1184): Make `@differentiable` on storage declarations propagate to
|
|
// the setter in addition to the getter.
|
|
@differentiable(reverse)
|
|
func protocolRequirementSetters<T: Protocol>(_ x: inout T, _ newValue: Float) {
|
|
// expected-error @+2 {{expression is not differentiable}}
|
|
// expected-note @+1 {{member is not differentiable because the corresponding protocol requirement is not '@differentiable'}}
|
|
x.property = newValue
|
|
// expected-error @+2 {{expression is not differentiable}}
|
|
// expected-note @+1 {{member is not differentiable because the corresponding protocol requirement is not '@differentiable'}}
|
|
x[] = newValue
|
|
}
|
|
|
|
// TF-1234: Test `@differentiable` propagation from class member storage
|
|
// declarations to their accessors in other file.
|
|
|
|
@differentiable(reverse)
|
|
func classRequirementGetters(_ x: Class) -> Float {
|
|
x.property + x[]
|
|
}
|
|
|
|
@differentiable(reverse)
|
|
func classRequirementSetters(_ x: inout Class, _ newValue: Float) {
|
|
x.property = newValue
|
|
x[] = newValue
|
|
}
|
|
|
|
// Test cross-file lookup of a derivative function with all-concrete derivative generic signature.
|
|
@differentiable(reverse)
|
|
func allConcreteDerivativeGenericSignature(_ a: [S]) -> Float {
|
|
// No error expected.
|
|
return a.sum()
|
|
}
|