Skip to content

Commit 73b0d8c

Browse files
author
zhangyue
committed
fix(tests): skip crashing torch ops on ascend
1 parent 2a2375a commit 73b0d8c

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

tests/test_torch_ops.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -208,6 +208,8 @@ def _list_default(aten_type):
208208
_VENDOR_CRASH_OPS = frozenset(
209209
{
210210
("npu", "mish"),
211+
("npu", "mse_loss"),
212+
("npu", "nonzero"),
211213
("npu", "nuclear_norm"),
212214
("npu", "_linalg_svd"),
213215
("npu", "svd"),

0 commit comments

Comments
 (0)