Skip to content

Commit 4fd9d09

Browse files
committed
real ad test
1 parent d4ebad4 commit 4fd9d09

File tree

1 file changed

+20
-21
lines changed

1 file changed

+20
-21
lines changed

lit/autodiff/square_real.thorin

+20-21
Original file line numberDiff line numberDiff line change
@@ -35,36 +35,35 @@
3535
ret b
3636
};
3737

38-
// .cn g [a:R32, ret: .Cn [R32, .Cn [R32, .Cn [R32]]]] = {
39-
// .let f_diff = %autodiff.autodiff (.Cn [R32,.Cn[R32]]) f;
40-
// f_diff (a,ret)
41-
// };
42-
43-
.cn .extern test [[], ret: .Cn [R32]] = {
44-
// .cn inner [r:R32, pb:.Cn [R32, .Cn [R32]]] = {
45-
// .cn pb_rec p::[R32] = {
46-
// // ret (5.0:R32)
47-
// ret r
48-
// };
49-
// pb (1.0:R32, pb_rec)
50-
// };
51-
// g (5.0:R32, inner)
38+
.cn g [a:R32, ret: .Cn [R32, .Cn [R32, .Cn [R32]]]] = {
39+
.let f_diff = %autodiff.autodiff (.Cn [R32,.Cn[R32]]) f;
40+
f_diff (a,ret)
41+
};
5242

43+
.cn .extern test [[], ret: .Cn [[R32,R32],[R32,R32]]] = {
44+
.cn inner [r:R32, pb:.Cn [R32, .Cn [R32]]] = {
45+
.cn pb_rec p::[R32] = {
46+
// ret (5.0:R32)
47+
ret ((25.0:R32,10.0:R32),(r,p))
48+
};
49+
pb (1.0:R32, pb_rec)
50+
};
51+
g (5.0:R32, inner)
5352

54-
// .cn pb_rec p::[R32] = {
55-
// ret p
56-
// };
53+
54+
// .cn pb_rec p::[R32] = {
55+
// ret (25.0:R32,p)
56+
// };
5757

5858
// f (5.0:R32, pb_rec)
5959

6060

6161
// ret (6.0:R32)
6262

6363
// .let b = %core.rop.mul (0, _32) (2.0:R32, 3.0:R32);
64-
.let b = 6.0:R32;
65-
ret b
64+
// .let b = 6.0:R32;
65+
// ret b
6666
};
6767

68-
// could return pair of (result, pb)
6968

70-
// CHECK-DAG: return{{.*}}176484
69+
// CHECK-DAG: ret{{.*}}‹2;{{.*}}›

0 commit comments

Comments
 (0)