Skip to content

Commit a44c7f8

Browse files
lukstafilukstafi
committed
Refactor DSLs into Operation.DSL_modules, formatting
1 parent cee3ab0 commit a44c7f8

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

48 files changed

+336
-367
lines changed

‎bin/compilation_speed.ml‎

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,7 @@ open Ocannl
33
module Tn = Ir.Tnode
44
module IDX = Train.IDX
55
module CDSL = Train.CDSL
6-
module TDSL = Operation.TDSL
7-
module NTDSL = Operation.NTDSL
8-
module PDSL = Operation.PDSL
6+
open Operation.DSL_modules
97

108
module type Backend = Ir.Backend_intf.Backend
119

‎bin/einsum_trivia.ml‎

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,7 @@ open Base
22
open Ocannl
33
module IDX = Train.IDX
44
module CDSL = Train.CDSL
5-
module TDSL = Operation.TDSL
6-
module NTDSL = Operation.NTDSL
7-
module PDSL = Operation.PDSL
5+
open Operation.DSL_modules
86

97
module type Backend = Ir.Backend_intf.Backend
108

‎bin/hello_world.ml‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,18 +2,18 @@ open Base
22
open Ocannl
33
module Tn = Ir.Tnode
44
module IDX = Train.IDX
5-
module TDSL = Operation.TDSL
6-
module NTDSL = Operation.NTDSL
7-
module PDSL = Operation.PDSL
5+
open Operation.DSL_modules
86
module CDSL = Train.CDSL
97

108
module type Backend = Ir.Backend_intf.Backend
119

1210
let hello1 () =
1311
let module Backend = (val Backends.fresh_backend ()) in
14-
let open Operation.TDSL in
12+
let open Ocannl.Operation.DSL_modules in
1513
(* Hey is inferred to be a matrix. *)
16-
let hey = range_of_shape ~batch_dims:[ 7 ] ~input_dims:[ 9; 10; 11 ] ~output_dims:[ 13; 14 ] () in
14+
let hey =
15+
TDSL.range_of_shape ~batch_dims:[ 7 ] ~input_dims:[ 9; 10; 11 ] ~output_dims:[ 13; 14 ] ()
16+
in
1717
let%op hoo = ((1 + 1) * hey) - 10 in
1818
(* For convenience, Train.forward will set hoo.value as fully on host. *)
1919
ignore (Train.forward_once (module Backend) hoo);

‎bin/hello_world_op.ml‎

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,7 @@ open Ocannl
33
module Tn = Ir.Tnode
44
module IDX = Train.IDX
55
module CDSL = Train.CDSL
6-
module TDSL = Operation.TDSL
7-
module PDSL = Operation.PDSL
6+
open Operation.DSL_modules
87

98
module type Backend = Ir.Backend_intf.Backend
109

‎bin/micrograd_basic.ml‎

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,7 @@ open Base
22
open Ocannl
33
module IDX = Train.IDX
44
module CDSL = Train.CDSL
5-
module TDSL = Operation.TDSL
6-
module NTDSL = Operation.NTDSL
7-
module PDSL = Operation.PDSL
5+
open Operation.DSL_modules
86

97
let _get_local_debug_runtime = Utils.get_local_debug_runtime
108

‎bin/micrograd_demo_logging.ml‎

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,7 @@
11
open Base
22
open Ocannl
33
module IDX = Train.IDX
4-
module TDSL = Operation.TDSL
5-
module NTDSL = Operation.NTDSL
6-
module PDSL = Operation.PDSL
4+
open Operation.DSL_modules
75
module CDSL = Train.CDSL
86
module Asgns = Ir.Assignments
97
module Tn = Ir.Tnode

‎bin/moons_benchmark.ml‎

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,7 @@ module Nd = Ir.Ndarray
44
module Ops = Ir.Ops
55
module Tn = Ir.Tnode
66
module IDX = Train.IDX
7-
module TDSL = Operation.TDSL
8-
module NTDSL = Operation.NTDSL
9-
module PDSL = Operation.PDSL
7+
open Operation.DSL_modules
108
module CDSL = Train.CDSL
119

1210
let _get_local_debug_runtime = Utils.get_local_debug_runtime

‎bin/primitive_ops.ml‎

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,7 @@ open Ocannl
33
module Tn = Ir.Tnode
44
module IDX = Train.IDX
55
module CDSL = Train.CDSL
6-
module TDSL = Operation.TDSL
7-
module NTDSL = Operation.NTDSL
8-
module PDSL = Operation.PDSL
6+
open Operation.DSL_modules
97

108
module type Backend = Ir.Backend_intf.Backend
119

‎bin/sanity_check.ml‎

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,7 @@ open! Base
22
open Ocannl
33
module Tn = Ir.Tnode
44
module IDX = Train.IDX
5-
module TDSL = Operation.TDSL
6-
module NTDSL = Operation.NTDSL
7-
module PDSL = Operation.PDSL
5+
open! Ocannl.Operation.DSL_modules
86
module CDSL = Train.CDSL
97

108
module type Backend = Ir.Backend_intf.Backend

‎bin/zero2hero_1of7.ml‎

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,7 @@ open Ocannl
33
module Tn = Ir.Tnode
44
module IDX = Train.IDX
55
module CDSL = Train.CDSL
6-
module TDSL = Operation.TDSL
7-
module NTDSL = Operation.NTDSL
8-
module PDSL = Operation.PDSL
6+
open Operation.DSL_modules
97

108
module type Backend = Ir.Backend_intf.Backend
119

0 commit comments

Comments
 (0)