Skip to content

Commit 9c206de

Browse files
authored
[MTHREADS]: enable resolve_neg op (#1234)
1 parent 3337a1d commit 9c206de

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

benchmark/test_special_perf.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,10 @@ def topk_input_fn(shape, dtype, device):
3030

3131
def resolve_neg_input_fn(shape, dtype, device):
3232
x = torch.randn(size=shape, dtype=dtype, device=device)
33-
yield x.conj().imag,
33+
if vendor_name == "mthreads":
34+
yield x.conj(),
35+
else:
36+
yield x.conj().imag,
3437

3538

3639
def resolve_conj_input_fn(shape, dtype, device):
@@ -61,8 +64,6 @@ def resolve_conj_input_fn(shape, dtype, device):
6164
],
6265
)
6366
def test_special_operations_benchmark(op_name, torch_op, dtypes, input_fn):
64-
if vendor_name == "mthreads" and op_name in ["resolve_neg"]:
65-
pytest.skip("Torch not supported complex")
6667
bench = GenericBenchmarkExcluse1D(
6768
input_fn=input_fn, op_name=op_name, dtypes=dtypes, torch_op=torch_op
6869
)

0 commit comments

Comments
 (0)