mirror of
https://github.com/apple/swift.git
synced 2025-12-21 12:14:44 +01:00
cddf73ecdb
Clean up a few general patterns that are now obviated by canImport This aligns more generally with the cleanup that the Swift Package Manager has already done in their automated XCTest-plumbing tool in apple/swift-package-manager#1826.
79 lines
2.3 KiB
Swift
79 lines
2.3 KiB
Swift
// RUN: %target-run-simple-swift
|
|
// REQUIRES: executable_test
|
|
|
|
import StdlibUnittest
|
|
#if canImport(Darwin)
|
|
import Darwin.C
|
|
#elseif canImport(Glibc)
|
|
import Glibc
|
|
#elseif os(Windows)
|
|
import ucrt
|
|
#else
|
|
#error("Unsupported platform")
|
|
#endif
|
|
import DifferentiationUnittest
|
|
|
|
var CustomDerivativesTests = TestSuite("CustomDerivatives")
|
|
|
|
// Specify non-differentiable functions.
|
|
// These will be wrapped in `differentiableFunction` and tested.
|
|
|
|
func unary(_ x: Tracked<Float>) -> Tracked<Float> {
|
|
var x = x
|
|
x *= 2
|
|
return x
|
|
}
|
|
|
|
func binary(_ x: Tracked<Float>, _ y: Tracked<Float>) -> Tracked<Float> {
|
|
var x = x
|
|
x *= y
|
|
return x
|
|
}
|
|
|
|
CustomDerivativesTests.testWithLeakChecking("differentiableFunction-unary") {
|
|
let diffableUnary = differentiableFunction { x in
|
|
(value: unary(x), pullback: { v in v * x * 2 })
|
|
}
|
|
expectEqual(20, gradient(at: 10, in: diffableUnary))
|
|
// Test differentiation of @differentiable function.
|
|
expectEqual(20, gradient(at: 10, in: { diffableUnary($0) }))
|
|
expectEqual(40, gradient(at: 10, in: { diffableUnary($0) * 2 }))
|
|
}
|
|
|
|
CustomDerivativesTests.testWithLeakChecking("differentiableFunction-binary") {
|
|
let diffableBinary = differentiableFunction { (x, y) in
|
|
(value: binary(x, y), pullback: { v in (v * y, v * x) })
|
|
}
|
|
expectEqual((10, 5), gradient(at: 5, 10, in: diffableBinary))
|
|
// Test differentiation of @differentiable function.
|
|
expectEqual((10, 5), gradient(at: 5, 10, in: { diffableBinary($0, $1) }))
|
|
expectEqual((20, 10), gradient(at: 5, 10, in: { diffableBinary($0, $1) * 2 }))
|
|
}
|
|
|
|
CustomDerivativesTests.testWithLeakChecking("SumOfGradPieces") {
|
|
var grad: Tracked<Float> = 0
|
|
func addToGrad(_ x: inout Tracked<Float>) { grad += x }
|
|
_ = gradient(at: 4) { (x: Tracked<Float>) in
|
|
x.withDerivative(addToGrad)
|
|
* x.withDerivative(addToGrad)
|
|
* x.withDerivative(addToGrad)
|
|
}
|
|
expectEqual(48, grad)
|
|
}
|
|
|
|
CustomDerivativesTests.testWithLeakChecking("ModifyGradientOfSum") {
|
|
expectEqual(30, gradient(at: 4) { (x: Tracked<Float>) in
|
|
x.withDerivative { $0 *= 10 } + x.withDerivative { $0 *= 20 }
|
|
})
|
|
}
|
|
|
|
CustomDerivativesTests.testWithLeakChecking("WithoutDerivative") {
|
|
expectEqual(0, gradient(at: Tracked<Float>(4)) { x in
|
|
withoutDerivative(at: x) { x in
|
|
Tracked<Float>(sinf(x.value) + cosf(x.value))
|
|
}
|
|
})
|
|
}
|
|
|
|
runAllTests()
|