Skip to content

Commit 2b58ad8

Browse files
committed
Experiment: use sync_cc backend in test/micrograd_demo
1 parent 2371568 commit 2b58ad8

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

test/micrograd_demo.ml

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,9 @@ let%expect_test "Micrograd README basic example" =
8282
let%expect_test "Micrograd half-moons example" =
8383
Tensor.unsafe_reinitialize ();
8484
Rand.init 5;
85-
let module Backend = (val Arrayjit.Backends.fresh_backend ()) in
85+
(* We could use the cc backend! But for unknown reason, it leads to different resuls on a
86+
lower_bounds CI target than on other CI targets. *)
87+
let module Backend = (val Arrayjit.Backends.fresh_backend ~backend_name:"sync_cc" ()) in
8688
let backend = (module Backend : Train.Backend_type with type context = Backend.context) in
8789
let device = Backend.(new_virtual_device @@ get_device ~ordinal:0) in
8890
let ctx = Backend.init device in

0 commit comments

Comments
 (0)