Skip to content

Commit 88e612c

Browse files
committed
feat: started einsum
1 parent 2829ddf commit 88e612c

File tree

5 files changed

+577
-0
lines changed

5 files changed

+577
-0
lines changed

.vscode/settings.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,7 @@
8383
"Brgemm",
8484
"cbnz",
8585
"einsum",
86+
"einsumtree",
8687
"errneous",
8788
"Fastor",
8889
"fmax",

CMakeLists.txt

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,8 @@ set(SRC_MAIN_FILES
9494
TensorOperation.cpp
9595
TensorOptimization.h
9696
TensorOptimization.cpp
97+
EinsumTree.h
98+
EinsumTree.cpp
9799
)
98100

99101
set(KERNEL_FILES
@@ -183,6 +185,7 @@ set(TEST_FILES
183185
Brgemm.test.cpp
184186
TensorOperation.test.cpp
185187
TensorOptimization.test.cpp
188+
EinsumTree.test.cpp
186189
)
187190

188191
set(TEST_KERNELS

0 commit comments

Comments
 (0)