Skip to content

Commit fb85e9c

Browse files
Disable paged optimizer test on Windows
1 parent 5b6a9b9 commit fb85e9c

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/test_optim.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import os
22
from os.path import join
33
import shutil
4+
import sys
45
import time
56
import uuid
67

@@ -168,6 +169,9 @@ def rm_path(path):
168169
@pytest.mark.parametrize("dim1", [1024], ids=id_formatter("dim1"))
169170
@pytest.mark.parametrize("dim2", [32, 1024, 4097, 1], ids=id_formatter("dim2"))
170171
def test_optimizer32bit(requires_cuda, dim1, dim2, gtype, optim_name):
172+
if optim_name.startswith("paged_") and sys.platform == "win32":
173+
pytest.skip("Paged optimizers can have issues on Windows.")
174+
171175
if gtype == torch.bfloat16 and optim_name in ["momentum", "rmsprop"]:
172176
pytest.skip()
173177
if dim1 == 1 and dim2 == 1:

0 commit comments

Comments
 (0)