Files
swift-mirror/test/AutoDiff/validation-test/existential.swift

47 lines
1.0 KiB
Swift

// RUN: %target-run-simple-swift
// REQUIRES: executable_test
import StdlibUnittest
import DifferentiationUnittest
var ExistentialTests = TestSuite("Existential")
protocol A {
@differentiable(reverse, wrt: x)
func a(_ x: Float) -> Float
}
func b(g: A) -> Float {
return gradient(at: 3) { x in g.a(x) }
}
struct B : A {
@differentiable(reverse, wrt: x)
func a(_ x: Float) -> Float { return x * 5 }
}
ExistentialTests.test("Existential method VJP-Tracked") {
expectEqual(5.0, b(g: B()))
}
/* Temporary disabled until https://github.com/swiftlang/swift/issues/84840 is fixed
We cannot use `Tracked<T>` :(
protocol ATracked {
@differentiable(reverse, wrt: x)
func a(_ x: Tracked<Float>) -> Tracked<Float>
}
func b(g: ATracked) -> Tracked<Float> {
return gradient(at: 3) { x in g.a(x) }
}
struct BTracked : ATracked {
@differentiable(reverse, wrt: x)
func a(_ x: Tracked<Float>) -> Tracked<Float> { return x * 5 }
}
ExistentialTests.testWithLeakChecking("Existential method VJP-Tracked") {
expectEqual(5.0, b(g: BTracked()))
}
*/
runAllTests()