mirror of
https://github.com/apple/swift.git
synced 2025-12-21 12:14:44 +01:00
0b53a02544
Rename the argument label `in:` in `gradient(at:in:)`, `pullback(at:in:)`, etc to `of:`, as suggested in the [pitch thread](https://forums.swift.org/t/differentiable-programming-for-gradient-based-machine-learning/42147).
20 lines
439 B
Swift
20 lines
439 B
Swift
// RUN: %target-run-simple-swift
|
|
// REQUIRES: executable_test
|
|
|
|
import StdlibUnittest
|
|
import DifferentiationUnittest
|
|
|
|
var RepeatedCallsTests = TestSuite("RepeatedCalls")
|
|
|
|
RepeatedCallsTests.testWithLeakChecking("Repeat") {
|
|
func mul2(_ x: Tracked<Float>) -> Tracked<Float> {
|
|
return 2 * x
|
|
}
|
|
func mul4(_ x: Tracked<Float>) -> Tracked<Float> {
|
|
return mul2(mul2(x))
|
|
}
|
|
expectEqual(4, gradient(at: 0, of: mul4))
|
|
}
|
|
|
|
runAllTests()
|