Skip to content

Commit f1ff552

Browse files
factnntengqm
authored andcommitted
Revert "【KernelGen】Add einsum operator (#1731)" (#3035)
This reverts commit e23da3c.
1 parent e94629c commit f1ff552

5 files changed

Lines changed: 0 additions & 801 deletions

File tree

benchmark/test_einsum.py

Lines changed: 0 additions & 192 deletions
This file was deleted.

src/flag_gems/__init__.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -195,7 +195,6 @@ def torch_ge(v):
195195
("divide_.Tensor", true_divide_),
196196
("divide_.Tensor_mode", div_mode_),
197197
("dot", dot),
198-
("einsum", einsum),
199198
("elu", elu),
200199
("elu_", elu_),
201200
("elu_backward", elu_backward),

src/flag_gems/ops/__init__.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,6 @@
109109
)
110110
from flag_gems.ops.dot import dot
111111
from flag_gems.ops.dropout import dropout, dropout_backward
112-
from flag_gems.ops.einsum import einsum
113112
from flag_gems.ops.elu import elu, elu_, elu_backward
114113
from flag_gems.ops.embedding import embedding, embedding_backward
115114
from flag_gems.ops.embedding_dense_backward import embedding_dense_backward
@@ -490,7 +489,6 @@
490489
"dot",
491490
"dropout",
492491
"dropout_backward",
493-
"einsum",
494492
"elu",
495493
"elu_",
496494
"elu_backward",

0 commit comments

Comments
 (0)