mirror of
https://github.com/apple/swift.git
synced 2025-12-21 12:14:44 +01:00
1be86adbfc
Delete `@differentiable` attribute `jvp:` and `vjp:` arguments for derivative registration. `@derivative` attribute is now the canonical way to register derivatives. Resolves TF-1001.
78 lines
2.2 KiB
Swift
78 lines
2.2 KiB
Swift
// RUN: rm -rf %t
|
|
// RUN: %swift-syntax-test -input-source-filename %s -parse-gen > %t
|
|
// RUN: diff -u %s %t
|
|
// RUN: %swift-syntax-test -input-source-filename %s -parse-gen -print-node-kind > %t.withkinds
|
|
// RUN: diff -u %S/Outputs/round_trip_parse_gen.swift.withkinds %t.withkinds
|
|
// RUN: %swift-syntax-test -input-source-filename %s -eof > %t
|
|
// RUN: diff -u %s %t
|
|
// RUN: %swift-syntax-test -serialize-raw-tree -input-source-filename %s > %t.dump
|
|
// RUN: %swift-syntax-test -deserialize-raw-tree -input-source-filename %t.dump -output-filename %t
|
|
// RUN: diff -u %s %t
|
|
|
|
// Note: RUN lines copied from test/Syntax/round_trip_parse_gen.swift.
|
|
|
|
@differentiable
|
|
func bar(_ x: Float, _: Float) -> Float { return 1 }
|
|
|
|
@differentiable(where T : FloatingPoint)
|
|
func bar<T : Numeric>(_ x: T, _: T) -> T { return 1 }
|
|
|
|
@differentiable(wrt: x)
|
|
func bar(_ x: Float, _: Float) -> Float { return 1 }
|
|
|
|
@differentiable(wrt: (self, x, y))
|
|
func bar(_ x: Float, y: Float) -> Float { return 1 }
|
|
|
|
@differentiable(wrt: (self, x, y) where T : FloatingPoint)
|
|
func bar<T : Numeric>(_ x: T, y: T) -> T { return 1 }
|
|
|
|
@derivative(of: -)
|
|
func negateDerivative(_ x: Float)
|
|
-> (value: Float, pullback: (Float) -> Float) {
|
|
return (-x, { v in -v })
|
|
}
|
|
|
|
@derivative(of: baz(label:_:), wrt: (x))
|
|
func bazDerivative(_ x: Float, y: Float)
|
|
-> (value: Float, pullback: (Float) -> Float) {
|
|
return (x, { v in v })
|
|
}
|
|
|
|
@transpose(of: -)
|
|
func negateDerivative(_ x: Float)
|
|
-> (value: Float, pullback: (Float) -> Float) {
|
|
return (-x, { v in -v })
|
|
}
|
|
|
|
@derivative(of: baz(label:_:), wrt: (x))
|
|
func bazDerivative(_ x: Float, y: Float)
|
|
-> (value: Float, pullback: (Float) -> Float) {
|
|
return (x, { v in v })
|
|
}
|
|
|
|
@derivative(of: A<T>.B<U, V>.C.foo(label:_:), wrt: x)
|
|
func qualifiedDerivative(_ x: Float, y: Float)
|
|
-> (value: Float, pullback: (Float) -> Float) {
|
|
return (x, { v in v })
|
|
}
|
|
|
|
@transpose(of: +)
|
|
func addTranspose(_ v: Float) -> (Float, Float) {
|
|
return (v, v)
|
|
}
|
|
|
|
@transpose(of: -, wrt: (0, 1))
|
|
func subtractTranspose(_ v: Float) -> (Float, Float) {
|
|
return (v, -v)
|
|
}
|
|
|
|
@transpose(of: Float.-, wrt: (0, 1))
|
|
func subtractTranspose(_ v: Float) -> (Float, Float) {
|
|
return (v, -v)
|
|
}
|
|
|
|
@derivative(of: A<T>.B<U, V>.C.foo(label:_:), wrt: 0)
|
|
func qualifiedTranspose(_ v: Float) -> (Float, Float) {
|
|
return (v, -v)
|
|
}
|