Tiny cleanup

ahrefs · May 29, 2023 · ae94068 · ae94068
1 parent b5b9e38
commit ae94068
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/lib/exec_as_gccjit.ml b/lib/exec_as_gccjit.ml
@@ -134,7 +134,7 @@ let get_tensor
                        RValue.int ctx c_index device_size_in_bytes;
                      ];
               if is_parallel || not update_on_host then
-                Block.eval (if is_parallel then task_finalize_block else replicated_finalize_block)
+                Block.eval (if is_replicated sync then replicated_finalize_block else task_finalize_block)
                 @@ RValue.call ctx (Function.builtin ctx "memcpy")
                      [
                        cast_void @@ LValue.address lhs;

diff --git a/test/zero2hero_1of7.ml b/test/zero2hero_1of7.ml
@@ -383,10 +383,10 @@ let%expect_test "2D neuron virtual" =
                         7.00e-1
                        Gradient
                        <void>
-                  [4] <*>                 │[1] <b>
-                   -6.00e+0               │ 6.70e+0
-                  Gradient                │Gradient
-                  <void>                  │<void>
+                   [4] <*>                │[1] <b>
+                   <void>                 │ 6.70e+0
+                   Gradient               │Gradient
+                   <void>                 │<void>
     [2] <w>            │[3] <x>           │
      -3.00e+0  1.00e+0 │ 2.00e+0  0.00e+0 │
     Gradient           │Gradient          │