Skip to content

Commit 980657f

Browse files
authored
[KUNLUNXIN] open elu and repeat_interleave_tensor in benchmark (flagos-ai#728)
* change for kunlunxin * format fix * open benchmark case * code format fix
1 parent 772be50 commit 980657f

2 files changed

Lines changed: 1 addition & 4 deletions

File tree

benchmark/test_tensor_concat_perf.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -197,7 +197,6 @@ def repeat_interleave_tensor_input_fn(shape, dtype, device):
197197
def test_tensor_repeat_benchmark(op_name, torch_op, input_fn, dtypes):
198198
if vendor_name == "kunlunxin" and op_name in [
199199
"repeat_interleave_self_tensor",
200-
"repeat_interleave_tensor",
201200
]:
202201
pytest.skip("RUNTIME TODOFIX")
203202
bench = TensorRepeatBenchmark(

benchmark/test_unary_pointwise_perf.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
FLOAT_DTYPES,
1313
INT_DTYPES,
1414
)
15-
from .performance_utils import Benchmark, generate_tensor_input, vendor_name
15+
from .performance_utils import Benchmark, generate_tensor_input
1616

1717

1818
class UnaryPointwiseBenchmark(Benchmark):
@@ -94,8 +94,6 @@ def get_tflops(self, op, *args, **kwargs):
9494
],
9595
)
9696
def test_general_unary_pointwise_perf(op_name, torch_op, dtypes):
97-
if vendor_name == "kunlunxin" and op_name == "elu":
98-
pytest.skip("RUNTIME TODOFIX")
9997
bench = UnaryPointwiseBenchmark(op_name=op_name, torch_op=torch_op, dtypes=dtypes)
10098
bench.run()
10199

0 commit comments

Comments
 (0)