diff --git a/tests/test_torch_ops.py b/tests/test_torch_ops.py index bc2a3337..1f0aff15 100644 --- a/tests/test_torch_ops.py +++ b/tests/test_torch_ops.py @@ -208,6 +208,8 @@ def _list_default(aten_type): _VENDOR_CRASH_OPS = frozenset( { ("npu", "mish"), + ("npu", "mse_loss"), + ("npu", "nonzero"), ("npu", "nuclear_norm"), ("npu", "_linalg_svd"), ("npu", "svd"),