Signed-off-by: Xiwen Yu <13230610+VALLIS-NERIA@users.noreply.github.com>
Signed-off-by: Mike Iovine <6158008+mikeiovine@users.noreply.github.com>
This commit is contained in:
xiweny 2025-10-17 16:13:01 +08:00 committed by Yanchao Lu
parent 6c8ba3be27
commit ce23e24123

View File

@ -655,7 +655,6 @@ def set_tensor_value_4(x, num_row, num_cols):
x.copy_(repeated)
@pytest.mark.skip(reason="https://nvbugs/5565565")
@skip_pre_blackwell
@pytest.mark.skipif(torch.cuda.device_count() < 4,
reason="needs 4 GPUs to run this test")