Skip to content

Commit 530087e

Browse files
lukstafilukstafi
committed
Update expectatiuons
1 parent 9f93509 commit 530087e

File tree

6 files changed

+313
-132
lines changed

6 files changed

+313
-132
lines changed

‎test/einsum/moons_demo_variant.expected‎

Lines changed: 11 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -24,22 +24,23 @@ n19 w2 as w2: Host&stream/412410; single prec 1x16; mem in bytes: <not-in-yet>
2424
n20 grad_w2 as w2.grad: Local/26046; single prec 1x16; mem in bytes: <not-in-yet>
2525
n21 @|_moons_input as moons_input: Virt/15; single prec 10x2; mem in bytes: <not-in-yet>
2626
n24 @|_moons_class as moons_class: Virt/15; single prec 10x1; mem in bytes: <not-in-yet>
27-
n27 * as n27: Local/1046; single prec 10x16; mem in bytes: <not-in-yet>
27+
n27 * as n27: Virt/151; single prec 10x16; mem in bytes: <not-in-yet>
2828
n28 grad_* as n27.grad: Local/1046; single prec 10x16; mem in bytes: <not-in-yet>
29-
n29 + as n29: Local/1046; single prec 10x16; mem in bytes: <not-in-yet>
30-
n30 grad_+ as n29.grad: Local/1046; single prec 10x16; mem in bytes: <not-in-yet>
29+
n29 + as n29: Virt/15; single prec 10x16; mem in bytes: <not-in-yet>
30+
n30 grad_+ as n29.grad: Virt/151; single prec 10x16; mem in bytes: <not-in-yet>
3131
n31 relu as relu: Local/1046; single prec 10x16; mem in bytes: <not-in-yet>
32-
n32 grad_relu as relu.grad: Local/1046; single prec 10x16; mem in bytes: <not-in-yet>
33-
n33 *_mlp_@|_moons_input as mlp_moons_input: Local/1046; single prec 10x1; mem in bytes: <not-in-yet>
32+
n32 grad_relu as relu.grad: Virt/151; single prec 10x16; mem in bytes: <not-in-yet>
33+
n33 *_mlp_@|_moons_input as mlp_moons_input: Virt/151; single prec 10x1; mem in bytes: <not-in-yet>
3434
n34 grad_*_mlp_@|_moons_input as mlp_moons_input.grad: Local/1046; single prec 10x1; mem in bytes: <not-in-yet>
3535
n35 *. as n35: Virt/15; single prec 10x1; mem in bytes: <not-in-yet>
36-
n36 grad_*. as n35.grad: Local/1046; single prec 10x1; mem in bytes: <not-in-yet>
36+
n36 grad_*. as n35.grad: Virt/151; single prec 10x1; mem in bytes: <not-in-yet>
3737
n37 1 as _1: Virt/40; single prec 1; mem in bytes: <not-in-yet>
38-
n38 - as n38: Local/1046; single prec 10x1; mem in bytes: <not-in-yet>
39-
n39 grad_- as n38.grad: Local/1046; single prec 10x1; mem in bytes: <not-in-yet>
38+
n38 - as n38: Virt/15; single prec 10x1; mem in bytes: <not-in-yet>
39+
n39 grad_- as n38.grad: Virt/151; single prec 10x1; mem in bytes: <not-in-yet>
4040
n40 relu_margin_loss as relu_margin_loss: Virt/15; single prec 10x1; mem in bytes: <not-in-yet>
41-
n41 grad_relu_margin_loss as relu_margin_loss.grad: Local/1046; single prec 10x1; mem in bytes: <not-in-yet>
41+
n41 grad_relu_margin_loss as relu_margin_loss.grad: Virt/151; single prec 10x1; mem in bytes: <not-in-yet>
4242
n42 10 as _10: Virt/40; single prec 1; mem in bytes: <not-in-yet>
43+
n43 => as n43: Local/9046; single prec 1; mem in bytes: <not-in-yet>
4344
n44 grad_=> as n43.grad: Virt/40; single prec 1; mem in bytes: <not-in-yet>
4445
n46 grad_/._scalar_loss as scalar_loss.grad: Virt/40; single prec 1; mem in bytes: <not-in-yet>
4546
n47 1 as _1: Virt/40; single prec 1; mem in bytes: <not-in-yet>
@@ -64,7 +65,7 @@ n66 sgd_momentum_w2 as sgd_momentum_w2: unknown; single prec <not-in-yet>; mem i
6465
n67 0.0001 as n67: Virt/40; single prec 1; mem in bytes: <not-in-yet>
6566
n68 *. as n68: Virt/15; single prec 1x16; mem in bytes: <not-in-yet>
6667
n69 point_mlp_result as point_mlp_result: Host&shared/37039; single prec 2; mem in bytes: <not-in-yet>
67-
n70 * as n70: Local/1046; single prec 16; mem in bytes: <not-in-yet>
68+
n70 * as n70: Virt/151; single prec 16; mem in bytes: <not-in-yet>
6869
n71 grad_* as n70.grad: unknown; single prec 16; mem in bytes: <not-in-yet>
6970
n72 + as n72: Virt/15; single prec 16; mem in bytes: <not-in-yet>
7071
n73 grad_+ as n72.grad: unknown; single prec 16; mem in bytes: <not-in-yet>

‎test/operations/hello_world_op.ml‎

Lines changed: 36 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -792,42 +792,42 @@ let%expect_test "Embed self id" =
792792
Train.printf ~here:[%here] ~with_code:false ~with_grad:false bar;
793793
[%expect
794794
{|
795-
HERE: test/operations/hello_world_op.ml:790:23
796-
┌─────────────────────────┐
797-
│[0]: !@self_id shape 0:1 │
798-
│┌┬──────┐ │
799-
│││axis 0│ │
800-
│├┼──────┤ │
801-
│││ 0.00 │ │
802-
│└┴──────┘ │
803-
└─────────────────────────┘
804-
HERE: test/operations/hello_world_op.ml:791:23
805-
┌─────────────────────────┐
806-
│[1]: !@self_id shape 0:1 │
807-
│┌┬──────┐ │
808-
│││axis 0│ │
809-
│├┼──────┤ │
810-
│││ 1.00 │ │
811-
│└┴──────┘ │
812-
└─────────────────────────┘
813-
HERE: test/operations/hello_world_op.ml:792:23
814-
┌─────────────────────┐
815-
│[4]: +_bar shape 0:1 │
816-
│┌┬──────┐ │
817-
│││axis 0│ │
818-
│├┼──────┤ │
819-
│││ 3.00 │ │
820-
│└┴──────┘ │
821-
└─────────────────────┘
822-
|}];
795+
HERE: test/operations/hello_world_op.ml:790:21
796+
┌─────────────────────────┐
797+
│[0]: !@self_id shape 0:1 │
798+
│┌┬──────┐ │
799+
│││axis 0│ │
800+
│├┼──────┤ │
801+
│││ 0.00 │ │
802+
│└┴──────┘ │
803+
└─────────────────────────┘
804+
HERE: test/operations/hello_world_op.ml:791:21
805+
┌─────────────────────────┐
806+
│[1]: !@self_id shape 0:1 │
807+
│┌┬──────┐ │
808+
│││axis 0│ │
809+
│├┼──────┤ │
810+
│││ 1.00 │ │
811+
│└┴──────┘ │
812+
└─────────────────────────┘
813+
HERE: test/operations/hello_world_op.ml:792:21
814+
┌─────────────────────┐
815+
│[4]: +_bar shape 0:1 │
816+
│┌┬──────┐ │
817+
│││axis 0│ │
818+
│├┼──────┤ │
819+
│││ 3.00 │ │
820+
│└┴──────┘ │
821+
└─────────────────────┘
822+
|}];
823823
Train.printf_tree ~here:[%here] bar;
824824
[%expect
825825
{|
826-
HERE: test/operations/hello_world_op.ml:822:30
827-
#4 +_bar
828-
3.00
829-
#3 + Virt/40 │#2 !@self_id Virt/40
830-
<void> │<void>
831-
#1 !@self_id│#0 !@self_id│
832-
1.00 │ 0.00 │
833-
|}]
826+
HERE: test/operations/hello_world_op.ml:823:26
827+
#4 +_bar
828+
3.00
829+
#3 + Virt/40 │#2 !@self_id Virt/40
830+
<void> │<void>
831+
#1 !@self_id│#0 !@self_id│
832+
1.00 │ 0.00 │
833+
|}]

0 commit comments

Comments
 (0)