mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
7cceaff5f3
Type annotations for instruction operands are omitted, e.g. ``` %3 = struct $S(%1, %2) ``` Operand types are redundant anyway and were only used for sanity checking in the SIL parser. But: operand types _are_ printed if the definition of the operand value was not printed yet. This happens: * if the block with the definition appears after the block where the operand's instruction is located * if a block or instruction is printed in isolation, e.g. in a debugger The old behavior can be restored with `-Xllvm -sil-print-types`. This option is added to many existing test files which check for operand types in their check-lines.
159 lines
8.9 KiB
Swift
159 lines
8.9 KiB
Swift
// RUN: %target-swift-frontend -Xllvm -sil-print-types -emit-silgen %s | %FileCheck %s
|
|
|
|
// Test derivative function vtable entries for `@differentiable` class members:
|
|
// - Methods.
|
|
// - Accessors (from properties and subscripts).
|
|
// - Initializers.
|
|
|
|
import _Differentiation
|
|
|
|
// Dummy `Differentiable`-conforming type.
|
|
struct DummyTangentVector: Differentiable & AdditiveArithmetic {
|
|
// FIXME(TF-648): Dummy to make `Super.TangentVector` be nontrivial.
|
|
var _nontrivial: [Float] = []
|
|
|
|
static var zero: Self { Self() }
|
|
static func + (_: Self, _: Self) -> Self { Self() }
|
|
static func - (_: Self, _: Self) -> Self { Self() }
|
|
typealias TangentVector = Self
|
|
}
|
|
|
|
class Super: Differentiable {
|
|
typealias TangentVector = DummyTangentVector
|
|
func move(by _: TangentVector) {}
|
|
|
|
var base: Float
|
|
// FIXME(TF-648): Dummy to make `Super.TangentVector` be nontrivial.
|
|
var _nontrivial: [Float] = []
|
|
|
|
init(base: Float) {
|
|
self.base = base
|
|
}
|
|
|
|
@differentiable(reverse, wrt: x)
|
|
func method(_ x: Float, _ y: Float) -> Float {
|
|
return x
|
|
}
|
|
|
|
@differentiable(reverse, wrt: x where T: Differentiable)
|
|
func genericMethod<T>(_ x: T, _ y: T) -> T {
|
|
return x
|
|
}
|
|
|
|
@differentiable(reverse)
|
|
var property: Float { base }
|
|
|
|
@differentiable(reverse, wrt: x)
|
|
subscript(_ x: Float, _ y: Float) -> Float {
|
|
return x
|
|
}
|
|
}
|
|
|
|
class Sub: Super {
|
|
override init(base: Float) {
|
|
super.init(base: base)
|
|
}
|
|
|
|
// Override JVP for `method` wrt `x`.
|
|
@derivative(of: method, wrt: x)
|
|
@derivative(of: subscript, wrt: x)
|
|
final func jvpMethod(_ x: Float, _ y: Float) -> (value: Float, differential: (Float) -> Float) {
|
|
fatalError()
|
|
}
|
|
// Override VJP for `method` wrt `x`.
|
|
@derivative(of: method, wrt: x)
|
|
@derivative(of: subscript, wrt: x)
|
|
final func vjpMethod(_ x: Float, _ y: Float) -> (value: Float, pullback: (Float) -> (Float)) {
|
|
fatalError()
|
|
}
|
|
|
|
// Override derivatives for `method` wrt `x`.
|
|
// FIXME(TF-1203): This `@differentiable` attribute should not be necessary to
|
|
// override derivatives. Fix `derivativeFunctionRequiresNewVTableEntry` to
|
|
// account for derived declaration `@derivative` attributes.
|
|
@differentiable(reverse, wrt: x)
|
|
// Add new derivatives for `method` wrt `(x, y)`.
|
|
@differentiable(reverse, wrt: (x, y))
|
|
override func method(_ x: Float, _ y: Float) -> Float {
|
|
return x
|
|
}
|
|
|
|
// Override derivatives for `property` wrt `self`.
|
|
@differentiable(reverse)
|
|
override var property: Float { base }
|
|
@derivative(of: property)
|
|
final func vjpProperty() -> (value: Float, pullback: (Float) -> TangentVector) {
|
|
fatalError()
|
|
}
|
|
|
|
// Override derivatives for `subscript` wrt `x`.
|
|
@differentiable(reverse, wrt: x)
|
|
override subscript(_ x: Float, _ y: Float) -> Float {
|
|
return x
|
|
}
|
|
}
|
|
|
|
class SubSub: Sub {}
|
|
|
|
// Check vtable entry thunks.
|
|
|
|
// CHECK-LABEL: sil private [transparent] [thunk] [ossa] @$s6vtable5SuperC6methodyS2f_SftFTJVfSUUpSr : $@convention(method) (Float, Float, @guaranteed Super) -> (Float, @owned @callee_guaranteed (Float) -> Float) {
|
|
// CHECK: bb0(%0 : $Float, %1 : $Float, %2 : @guaranteed $Super):
|
|
// CHECK: %3 = function_ref @$s6vtable5SuperC6methodyS2f_SftF : $@convention(method) (Float, Float, @guaranteed Super) -> Float
|
|
// CHECK: %4 = differentiable_function [parameters 0] [results 0] %3 : $@convention(method) (Float, Float, @guaranteed Super) -> Float
|
|
// CHECK: %5 = differentiable_function_extract [jvp] %4 : $@differentiable(reverse) @convention(method) (Float, @noDerivative Float, @noDerivative @guaranteed Super) -> Float
|
|
// CHECK: %6 = apply %5(%0, %1, %2) : $@convention(method) (Float, Float, @guaranteed Super) -> (Float, @owned @callee_guaranteed (Float) -> Float)
|
|
// CHECK: return %6 : $(Float, @callee_guaranteed (Float) -> Float)
|
|
// CHECK: }
|
|
|
|
// Check vtable entries: new vs `[override]` vs `[inherited]` entries.
|
|
|
|
// CHECK-LABEL: sil_vtable Super {
|
|
// CHECK: #Super.method: (Super) -> (Float, Float) -> Float : @$s6vtable5SuperC6methodyS2f_SftF
|
|
// CHECK: #Super.method!jvp.SUU: (Super) -> (Float, Float) -> Float : @$s6vtable5SuperC6methodyS2f_SftFTJVfSUUpSr
|
|
// CHECK: #Super.method!vjp.SUU: (Super) -> (Float, Float) -> Float : @$s6vtable5SuperC6methodyS2f_SftFTJVrSUUpSr
|
|
// CHECK: #Super.genericMethod: <T> (Super) -> (T, T) -> T : @$s6vtable5SuperC13genericMethodyxx_xtlF
|
|
// CHECK: #Super.genericMethod!jvp.SUU.<T where T : Differentiable>: <T> (Super) -> (T, T) -> T : @$s6vtable5SuperC13genericMethodyxx_xtlF16_Differentiation14DifferentiableRzlTJVfSUUpSr
|
|
// CHECK: #Super.genericMethod!vjp.SUU.<T where T : Differentiable>: <T> (Super) -> (T, T) -> T : @$s6vtable5SuperC13genericMethodyxx_xtlF16_Differentiation14DifferentiableRzlTJVrSUUpSr
|
|
// CHECK: #Super.property!getter: (Super) -> () -> Float : @$s6vtable5SuperC8propertySfvg
|
|
// CHECK: #Super.property!getter.jvp.S: (Super) -> () -> Float : @$s6vtable5SuperC8propertySfvgTJVfSpSr
|
|
// CHECK: #Super.property!getter.vjp.S: (Super) -> () -> Float : @$s6vtable5SuperC8propertySfvgTJVrSpSr
|
|
// CHECK: #Super.subscript!getter: (Super) -> (Float, Float) -> Float : @$s6vtable5SuperCyS2f_Sftcig
|
|
// CHECK: #Super.subscript!getter.jvp.SUU: (Super) -> (Float, Float) -> Float : @$s6vtable5SuperCyS2f_SftcigTJVfSUUpSr
|
|
// CHECK: #Super.subscript!getter.vjp.SUU: (Super) -> (Float, Float) -> Float : @$s6vtable5SuperCyS2f_SftcigTJVrSUUpSr
|
|
// CHECK: }
|
|
|
|
// CHECK-LABEL: sil_vtable Sub {
|
|
// CHECK: #Super.method: (Super) -> (Float, Float) -> Float : @$s6vtable3SubC6methodyS2f_SftF [override]
|
|
// CHECK: #Super.method!jvp.SUU: (Super) -> (Float, Float) -> Float : @$s6vtable3SubC6methodyS2f_SftFTJVfSUUpSr [override]
|
|
// CHECK: #Super.method!vjp.SUU: (Super) -> (Float, Float) -> Float : @$s6vtable3SubC6methodyS2f_SftFTJVrSUUpSr [override]
|
|
// CHECK: #Super.genericMethod: <T> (Super) -> (T, T) -> T : @$s6vtable5SuperC13genericMethodyxx_xtlF [inherited]
|
|
// CHECK: #Super.genericMethod!jvp.SUU.<T where T : Differentiable>: <T> (Super) -> (T, T) -> T : @$s6vtable5SuperC13genericMethodyxx_xtlF16_Differentiation14DifferentiableRzlTJVfSUUpSr [inherited]
|
|
// CHECK: #Super.genericMethod!vjp.SUU.<T where T : Differentiable>: <T> (Super) -> (T, T) -> T : @$s6vtable5SuperC13genericMethodyxx_xtlF16_Differentiation14DifferentiableRzlTJVrSUUpSr [inherited]
|
|
// CHECK: #Super.property!getter: (Super) -> () -> Float : @$s6vtable3SubC8propertySfvg [override]
|
|
// CHECK: #Super.property!getter.jvp.S: (Super) -> () -> Float : @$s6vtable3SubC8propertySfvgTJVfSpSr [override]
|
|
// CHECK: #Super.property!getter.vjp.S: (Super) -> () -> Float : @$s6vtable3SubC8propertySfvgTJVrSpSr [override]
|
|
// CHECK: #Super.subscript!getter: (Super) -> (Float, Float) -> Float : @$s6vtable3SubCyS2f_Sftcig [override]
|
|
// CHECK: #Super.subscript!getter.jvp.SUU: (Super) -> (Float, Float) -> Float : @$s6vtable3SubCyS2f_SftcigTJVfSUUpSr [override]
|
|
// CHECK: #Super.subscript!getter.vjp.SUU: (Super) -> (Float, Float) -> Float : @$s6vtable3SubCyS2f_SftcigTJVrSUUpSr [override]
|
|
// CHECK: #Sub.method!jvp.SSU: (Sub) -> (Float, Float) -> Float : @$s6vtable3SubC6methodyS2f_SftFTJVfSSUpSr
|
|
// CHECK: #Sub.method!vjp.SSU: (Sub) -> (Float, Float) -> Float : @$s6vtable3SubC6methodyS2f_SftFTJVrSSUpSr
|
|
// CHECK: }
|
|
|
|
// CHECK-LABEL: sil_vtable SubSub {
|
|
// CHECK: #Super.method: (Super) -> (Float, Float) -> Float : @$s6vtable3SubC6methodyS2f_SftF [inherited]
|
|
// CHECK: #Super.method!jvp.SUU: (Super) -> (Float, Float) -> Float : @$s6vtable3SubC6methodyS2f_SftFTJVfSUUpSr [inherited]
|
|
// CHECK: #Super.method!vjp.SUU: (Super) -> (Float, Float) -> Float : @$s6vtable3SubC6methodyS2f_SftFTJVrSUUpSr [inherited]
|
|
// CHECK: #Super.genericMethod: <T> (Super) -> (T, T) -> T : @$s6vtable5SuperC13genericMethodyxx_xtlF [inherited]
|
|
// CHECK: #Super.genericMethod!jvp.SUU.<T where T : Differentiable>: <T> (Super) -> (T, T) -> T : @$s6vtable5SuperC13genericMethodyxx_xtlF16_Differentiation14DifferentiableRzlTJVfSUUpSr [inherited]
|
|
// CHECK: #Super.genericMethod!vjp.SUU.<T where T : Differentiable>: <T> (Super) -> (T, T) -> T : @$s6vtable5SuperC13genericMethodyxx_xtlF16_Differentiation14DifferentiableRzlTJVrSUUpSr [inherited]
|
|
// CHECK: #Super.property!getter: (Super) -> () -> Float : @$s6vtable3SubC8propertySfvg [inherited]
|
|
// CHECK: #Super.property!getter.jvp.S: (Super) -> () -> Float : @$s6vtable3SubC8propertySfvgTJVfSpSr [inherited]
|
|
// CHECK: #Super.property!getter.vjp.S: (Super) -> () -> Float : @$s6vtable3SubC8propertySfvgTJVrSpSr [inherited]
|
|
// CHECK: #Super.subscript!getter: (Super) -> (Float, Float) -> Float : @$s6vtable3SubCyS2f_Sftcig [inherited]
|
|
// CHECK: #Super.subscript!getter.jvp.SUU: (Super) -> (Float, Float) -> Float : @$s6vtable3SubCyS2f_SftcigTJVfSUUpSr [inherited]
|
|
// CHECK: #Super.subscript!getter.vjp.SUU: (Super) -> (Float, Float) -> Float : @$s6vtable3SubCyS2f_SftcigTJVrSUUpSr [inherited]
|
|
// CHECK: #Sub.method!jvp.SSU: (Sub) -> (Float, Float) -> Float : @$s6vtable3SubC6methodyS2f_SftFTJVfSSUpSr [inherited]
|
|
// CHECK: #Sub.method!vjp.SSU: (Sub) -> (Float, Float) -> Float : @$s6vtable3SubC6methodyS2f_SftFTJVrSSUpSr [inherited]
|
|
// CHECK: }
|