Skip to content

Commit

Permalink
Update
Browse files Browse the repository at this point in the history
[ghstack-poisoned]
  • Loading branch information
vmoens committed Oct 31, 2024
1 parent 02aa6ed commit 5467cbb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion benchmarks/common/h2d_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -150,7 +150,7 @@ def consolidate(td, num_threads):
],
)
@pytest.mark.skipif(
TORCH_VERSION < version.parse("2.5.1"), reason="requires torch>=2.5"
TORCH_VERSION < version.parse("2.5.2"), reason="requires torch>=2.5"
)
@pytest.mark.skipif(not torch.cuda.is_available(), reason="no CUDA device found")
class TestTo:
Expand Down

0 comments on commit 5467cbb

Please sign in to comment.