We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2371568 commit 2b58ad8Copy full SHA for 2b58ad8
test/micrograd_demo.ml
@@ -82,7 +82,9 @@ let%expect_test "Micrograd README basic example" =
82
let%expect_test "Micrograd half-moons example" =
83
Tensor.unsafe_reinitialize ();
84
Rand.init 5;
85
- let module Backend = (val Arrayjit.Backends.fresh_backend ()) in
+ (* 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
88
let backend = (module Backend : Train.Backend_type with type context = Backend.context) in
89
let device = Backend.(new_virtual_device @@ get_device ~ordinal:0) in
90
let ctx = Backend.init device in
0 commit comments