Updated test for embed self id

lukstafi · lukstafi · commit 43cc8d2b0222 · 2025-07-29T15:50:46.000+02:00
diff --git a/test/operations/hello_world_op.ml b/test/operations/hello_world_op.ml
@@ -782,18 +782,16 @@ let%expect_test "Very big tensor" =
     in
     let%op hey = embed_self_id () in
     let%op hoo = embed_self_id () in
-    (* let%op bar = hoo + hey + embed_self_id () in *)
+    let%op bar = hoo + hey + embed_self_id () in
     Train.set_hosted hey.value;
     Train.set_hosted hoo.value;
-    (* Train.set_hosted bar.value; *)
-    ignore (Train.forward_once backend hey);
-    ignore (Train.forward_once backend hoo);
-    (* ignore (Train.forward_once backend bar); *)
+    Train.set_hosted bar.value;
+    ignore (Train.forward_once backend bar);
     Train.printf ~here:[%here] ~with_code:false ~with_grad:false hey;
     Train.printf ~here:[%here] ~with_code:false ~with_grad:false hoo;
-    (* Train.printf ~here:[%here] ~with_code:false ~with_grad:false bar; *)
+    Train.printf ~here:[%here] ~with_code:false ~with_grad:false bar;
     [%expect {|
-      HERE: test/operations/hello_world_op.ml:792:23
+      HERE: test/operations/hello_world_op.ml:790:23
       ┌─────────────────────────┐
       │[0]: !@self_id shape 0:1 │
       │┌┬──────┐                │
@@ -802,7 +800,7 @@ let%expect_test "Very big tensor" =
       │││ 0.00 │                │
       │└┴──────┘                │
       └─────────────────────────┘
-      HERE: test/operations/hello_world_op.ml:793:23
+      HERE: test/operations/hello_world_op.ml:791:23
       ┌─────────────────────────┐
       │[1]: !@self_id shape 0:1 │
       │┌┬──────┐                │
@@ -811,5 +809,24 @@ let%expect_test "Very big tensor" =
       │││ 1.00 │                │
       │└┴──────┘                │
       └─────────────────────────┘
-      |}]
+      HERE: test/operations/hello_world_op.ml:792:23
+      ┌─────────────────────┐
+      │[4]: +_bar shape 0:1 │
+      │┌┬──────┐            │
+      │││axis 0│            │
+      │├┼──────┤            │
+      │││ 3.00 │            │
+      │└┴──────┘            │
+      └─────────────────────┘
+      |}];
+      Train.printf_tree ~here:[%here] bar;
+      [%expect {|
+        HERE: test/operations/hello_world_op.ml:822:30
+                           #4 +_bar
+                            3.00
+              #3 + Virt/40       │#2 !@self_id Virt/40
+              <void>             │<void>
+        #1 !@self_id│#0 !@self_id│
+         1.00       │ 0.00       │
+        |}]
     
