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

31 lines
1.3 KiB
Swift

// RUN: %empty-directory(%t)
// RUN: %target-build-swift-dylib(%t/%target-library-name(cross_module_differentiation_other)) %S/Inputs/cross_module_differentiation_other.swift -emit-module -emit-module-path %t/cross_module_differentiation_other.swiftmodule -module-name cross_module_differentiation_other
// RUN: %target-build-swift -I%t -L%t %s -o %t/a.out -lcross_module_differentiation_other %target-rpath(%t)
// RUN: %target-codesign %t/a.out
// RUN: %target-codesign %t/%target-library-name(cross_module_differentiation_other)
// RUN: %target-run %t/a.out %t/%target-library-name(cross_module_differentiation_other)
// REQUIRES: executable_test
// TF-1025: Test differentiability witness linkage for `PublicNonABI` original functions.
// TF-1239: Test `SynthesizedFileUnit` TBDGen.
import cross_module_differentiation_other
import _Differentiation
import StdlibUnittest
var CrossModuleTests = TestSuite("E2ECrossModule")
CrossModuleTests.test("differentiable function default argument") {
let actualGrad = gradient(at: 0) { applyArgument($0) }
let expectedGrad: Float = 1
expectEqual(actualGrad, expectedGrad)
}
CrossModuleTests.test("differentiable function specified default argument") {
let actualGrad = gradient(at: 0) { applyArgument($0, { $0 }) }
let expectedGrad: Float = 1
expectEqual(actualGrad, expectedGrad)
}
runAllTests()