Skip to content

Commit 334b011

Browse files
authored
Updated test_ops.py to skip non_standard_bool_values tests (ROCm#38)
1 parent c78f91c commit 334b011

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

test/test_ops.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1434,6 +1434,19 @@ def test_complex_half_reference_testing(self, device, dtype, op):
14341434

14351435
@ops(op_db, allowed_dtypes=(torch.bool,))
14361436
def test_non_standard_bool_values(self, device, dtype, op):
1437+
if TEST_WITH_ROCM and "cuda" in device:
1438+
rocm_blocklist = [
1439+
"test_non_standard_bool_values_masked_scatter_cuda_bool",
1440+
"test_non_standard_bool_values_nn_functional_unfold_cuda_bool", # only in rocm6.4_internal_testing
1441+
"test_non_standard_bool_values_scatter_add_cuda_bool",
1442+
"test_non_standard_bool_values_scatter_cuda_bool",
1443+
"test_non_standard_bool_values_scatter_reduce_sum_cuda_bool",
1444+
"test_non_standard_bool_values_tril_cuda_bool",
1445+
"test_non_standard_bool_values_triu_cuda_bool",
1446+
]
1447+
if self._testMethodName in rocm_blocklist:
1448+
self.skipTest("Failed on ROCm")
1449+
14371450
# Test boolean values other than 0x00 and 0x01 (gh-54789)
14381451
def convert_boolean_tensors(x):
14391452
if not isinstance(x, torch.Tensor) or x.dtype != torch.bool:

0 commit comments

Comments
 (0)