Skip to content

Commit da1dad5

Browse files
committed
remove bfloat16 type from test_tensor_atomic_add_shift_1 test as unsupported type for atomic op
Signed-off-by: Anatoly Myachev <[email protected]>
1 parent f88edfa commit da1dad5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/test/unit/language/test_core.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1755,7 +1755,7 @@ def kernel(X, val, NUM: tl.constexpr):
17551755
@pytest.mark.parametrize("size, num_ctas, dtype_x_str", [(size, num_ctas, dtype_x_str)
17561756
for size in [2, 4, 8, 32, 64, 128]
17571757
for num_ctas in num_ctas_list
1758-
for dtype_x_str in ['bfloat16', 'float16', 'float32']])
1758+
for dtype_x_str in ['float16', 'float32']])
17591759
def test_tensor_atomic_add_shift_1(size, num_ctas, dtype_x_str, device):
17601760
check_type_supported(dtype_x_str, device)
17611761

0 commit comments

Comments
 (0)