Skip to content

Commit eea2724

Browse files
committed
Remove printing range_of_shape tensors in tests
(they are virtual now by default).
1 parent 88cd3bc commit eea2724

File tree

6 files changed

+1
-630
lines changed

6 files changed

+1
-630
lines changed

bin/einsum_trivia.ml

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -66,12 +66,9 @@ let () =
6666
let a = TDSL.range_of_shape ~batch_dims:[ 2 ] ~input_dims:[ 3 ] ~output_dims:[ 4 ] () in
6767
let b = TDSL.range_of_shape ~batch_dims:[ 2 ] ~input_dims:[ 4 ] ~output_dims:[ 5 ] () in
6868
let%op a2 = a *+ "b|i->o; b|i->o => b|i->o" a in
69-
Tensor.print ~spy:true ~with_code:false ~with_grad:false `Default @@ a;
7069
let ctx = Utils.capture_stdout_logs (fun () -> Train.forward_and_ctx backend ctx a2) in
71-
Tensor.print ~here:[%here] ~with_code:false ~with_grad:false `Default @@ a;
7270
let%op c = b *+ "b|h->o; b|i->h => b|i->o" a in
7371
Utils.capture_stdout_logs (fun () -> Train.forward_and_forget backend ctx c);
74-
Tensor.print ~here:[%here] ~with_code:false ~with_grad:false `Default @@ a;
7572
(* let%op d = a *+ "a|i->h; b|h->o => ab|i->o" b in Utils.capture_stdout_logs (fun () ->
7673
Train.forward_and_forget backend ctx d); let%op e = a *+ "b|i->h; b|h->o => i->o" b in
7774
Utils.capture_stdout_logs (fun () -> Train.forward_and_forget backend ctx e); let%op f = a *+

bin/hello_world.ml

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -55,9 +55,6 @@ let hello3 () =
5555
let y = TDSL.O.(( + ) ~label:[ "y" ] (hey * zero_to_twenty) zero_to_twenty) in
5656
Train.set_hosted hey.value;
5757
let routine = Train.to_routine (module Backend) ctx IDX.empty @@ Train.forward y in
58-
Tensor.print ~here:[%here] ~with_code:true ~with_grad:false `Inline zero_to_twenty;
59-
Tensor.print ~here:[%here] ~with_code:true ~with_grad:false `Default zero_to_twenty;
60-
Tensor.print_tree ~with_grad:false ~depth:9 zero_to_twenty;
6158
Stdio.printf "\n%!";
6259
Train.run routine;
6360
Tensor.print ~here:[%here] ~with_code:true ~with_grad:false `Default y;
@@ -117,7 +114,6 @@ let hello5 () =
117114
let hey = TDSL.range_of_shape ~batch_dims:[ 2 ] ~input_dims:[ 3 ] ~output_dims:[ 4 ] () in
118115
let%op ho = hey ++ "...|1->... => ...|..." in
119116
Train.forward_and_forget backend ctx ho;
120-
Tensor.print ~here:[%here] ~with_code:false ~with_grad:false `Default @@ hey;
121117
Tensor.print ~here:[%here] ~with_code:false ~with_grad:false `Default @@ ho
122118

123119
let hello6 () =

bin/hello_world_op.ml

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -193,8 +193,6 @@ let%track2_sexp _Big_matrix (() : unit) : unit =
193193
let zero_to_twenty = TDSL.range 20 in
194194
let%op yd = (hey * zero_to_twenty) + zero_to_twenty in
195195
Train.forward_and_forget backend ctx yd;
196-
Tensor.print ~here:[%here] ~with_code:false ~with_grad:false `Inline zero_to_twenty;
197-
Tensor.print ~here:[%here] ~with_code:false ~with_grad:false `Default zero_to_twenty;
198196
Tensor.print ~here:[%here] ~with_code:false ~with_grad:false `Default hey;
199197
Tensor.print ~here:[%here] ~with_code:false ~with_grad:false `Default yd
200198

@@ -215,7 +213,6 @@ let%track2_sexp _Very_big_tensor (() : unit) : unit =
215213
let hey = TDSL.range_of_shape ~batch_dims:[ 6 ] ~input_dims:[ 7; 8 ] ~output_dims:[ 9 ] () in
216214
let%op ye = (hey * (1 + 1)) - 10 in
217215
Train.forward_and_forget backend ctx ye;
218-
Tensor.print ~here:[%here] ~with_code:false ~with_grad:false `Default hey;
219216
Tensor.print ~here:[%here] ~with_code:false ~with_grad:false `Default ye
220217

221218
let _suspended (() : unit) : unit =

0 commit comments

Comments
 (0)