Skip to content

Commit 3b27971

Browse files
[Test] Drop default device value in clamping downcast test (#7387)
This change removes the default `cuda` value from the `device` argument in `test_typeconvert_downcast_clamping`. Giving it a default value prevents you from setting it via the commandline.
1 parent ec8cb09 commit 3b27971

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/test/unit/language/test_conversions.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -369,7 +369,7 @@ def test_typeconvert_downcast(src_dtype, dst_dtype, rounding, max_repr, device):
369369
])
370370
@pytest.mark.parametrize("dst_dtype", ["float8e4nv", "float8e5"])
371371
@pytest.mark.parametrize("src_dtype", ["float32", "float16", "bfloat16"])
372-
def test_typeconvert_downcast_clamping(src_dtype, dst_dtype, mode, rounding="rtne", device="cuda"):
372+
def test_typeconvert_downcast_clamping(src_dtype, dst_dtype, mode, device, rounding="rtne"):
373373
if is_cuda():
374374
if src_dtype != 'float32' and torch.cuda.get_device_capability(0) < (9, 0):
375375
pytest.skip("non-float32 downcast tests only supported on NVGPU with compute capability 9.0+")

0 commit comments

Comments
 (0)