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.
55 lines
2.2 KiB
Swift
55 lines
2.2 KiB
Swift
// RUN: %target-build-swift %s
|
|
// RUN: %target-swift-frontend -emit-sil %s | %FileCheck %s
|
|
|
|
// Test crashes related to differentiation and definite intiialization.
|
|
|
|
// SR-12886: SIL memory lifetime verification error due to
|
|
// `SILCloner::visitAllocStack` not copying the `[dynamic_lifetime]` attribute.
|
|
|
|
// SR-12887: Debug scope error for pullback struct `struct` instruction
|
|
// generated by `VJPEmitter`.
|
|
|
|
// FIXME(SR-13021): Disabled due to flakiness on Linux, likely related to TF-1197.
|
|
// REQUIRES: SR13021
|
|
|
|
import _Differentiation
|
|
|
|
enum Enum {
|
|
case a
|
|
}
|
|
|
|
struct Tensor<T>: Differentiable {
|
|
@noDerivative var x: T
|
|
@noDerivative var optional: Int?
|
|
|
|
init(_ x: T, _ e: Enum) {
|
|
self.x = x
|
|
switch e {
|
|
case .a: optional = 1
|
|
}
|
|
}
|
|
|
|
// Definite initialization triggers for this initializer.
|
|
@differentiable(reverse)
|
|
init(_ x: T, _ other: Self) {
|
|
self = Self(x, Enum.a)
|
|
}
|
|
}
|
|
|
|
// Check that `allock_stack [dynamic_lifetime]` attribute is correctly cloned.
|
|
|
|
// CHECK-LABEL: sil hidden @$s4main6TensorVyACyxGx_ADtcfC : $@convention(method) <T> (@in T, @in Tensor<T>, @thin Tensor<T>.Type) -> @out Tensor<T> {
|
|
// CHECK: [[SELF_ALLOC:%.*]] = alloc_stack [dynamic_lifetime] $Tensor<T>, var, name "self"
|
|
|
|
// CHECK-LABEL: sil hidden @AD__$s4main6TensorVyACyxGx_ADtcfC__vjp_src_0_wrt_1_l : $@convention(method) <τ_0_0> (@in τ_0_0, @in Tensor<τ_0_0>, @thin Tensor<τ_0_0>.Type) -> (@out Tensor<τ_0_0>, @owned @callee_guaranteed @substituted <τ_0_0, τ_0_1> (@in_guaranteed τ_0_0) -> @out τ_0_1 for <Tensor<τ_0_0>.TangentVector, Tensor<τ_0_0>.TangentVector>) {
|
|
// CHECK: [[SELF_ALLOC:%.*]] = alloc_stack [dynamic_lifetime] $Tensor<τ_0_0>, var, name "self"
|
|
|
|
// SR-12886 original error:
|
|
// SIL memory lifetime failure in @AD__$s5crash6TensorVyACyxGx_ADtcfC__vjp_src_0_wrt_1_l: memory is not initialized, but should
|
|
// memory location: %29 = struct_element_addr %5 : $*Tensor<τ_0_0>, #Tensor.x // user: %30
|
|
// at instruction: destroy_addr %29 : $*τ_0_0 // id: %30
|
|
|
|
// SR-12887 original error:
|
|
// SIL verification failed: Basic block contains a non-contiguous lexical scope at -Onone: DS == LastSeenScope
|
|
// %26 = struct $_AD__$s5crash6TensorVyACyxGx_ADtcfC_bb0__PB__src_0_wrt_1_l<τ_0_0> () // users: %34, %28
|