1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
// RUN: %target-run-simple-swift
// REQUIRES: executable_test
// This test fails on arm64e with a pointer auth failure.
// XFAIL: CPU=arm64e
import _Differentiation
import StdlibUnittest
var ModifyAccessorTests = TestSuite("ModifyAccessor")
ModifyAccessorTests.test("SimpleModifyAccessor") {
struct S: Differentiable {
private var _x : Float
func _endMutation() {}
var x: Float {
get{_x}
set(newValue) { _x = newValue }
_modify {
defer { _endMutation() }
if (x > 0) {
yield &_x
} else {
yield &_x
}
}
}
init(_ x : Float) {
self._x = x
}
}
func modify_struct(_ x : Float) -> Float {
var s = S(x)
s.x *= s.x
return s.x
}
expectEqual((100, 20), valueWithGradient(at: 10, of: modify_struct))
}
ModifyAccessorTests.test("GenericModifyAccessor") {
struct S<T : Differentiable & SignedNumeric & Comparable>: Differentiable {
private var _x : T
func _endMutation() {}
var x: T {
get{_x}
set(newValue) { _x = newValue }
_modify {
defer { _endMutation() }
if (x > -x) {
yield &_x
} else {
yield &_x
}
}
}
init(_ x : T) {
self._x = x
}
}
func modify_struct(_ x : Float) -> Float {
var s = S<Float>(x)
s.x *= s.x
return s.x
}
expectEqual((100, 20), valueWithGradient(at: 10, of: modify_struct))
}
runAllTests()
|
