Skip to content

Commit 0d02815

Browse files
Skip test on torch==2.8.0+cpu for Windows regression
1 parent 2ffbf22 commit 0d02815

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

tests/test_functional.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,10 @@
11
import math
2+
import platform
23
import random
34
import time
45

56
import einops
7+
from packaging import version
68
import pytest
79
import torch
810

@@ -1407,7 +1409,7 @@ def test_gemv_4bit(self, device, dim, dtype, storage_type, quant_storage, double
14071409
@pytest.mark.parametrize("dtype", [torch.float16, torch.bfloat16, torch.float32], ids=describe_dtype)
14081410
@pytest.mark.skipif(
14091411
HIP_ENVIRONMENT and ROCM_GPU_ARCH == "gfx90a",
1410-
reason="this test is not supported on ROCm with gfx90a architectßure yet",
1412+
reason="this test is not supported on ROCm with gfx90a architecture yet",
14111413
)
14121414
def test_gemv_eye_4bit(self, device, storage_type, dtype):
14131415
if device == "cpu" and dtype == torch.bfloat16 and torch.__version__ < (2, 3):
@@ -1416,6 +1418,9 @@ def test_gemv_eye_4bit(self, device, storage_type, dtype):
14161418
if device == "hpu" and not is_supported_on_hpu(storage_type, dtype):
14171419
pytest.skip("This configuration is not supported on HPU.")
14181420

1421+
if device == "cpu" and platform.system() == "Windows" and version.parse(torch.__version__.release) == (2, 8, 0):
1422+
pytest.skip("Regression: CPU crash on Windows with torch 2.8.0")
1423+
14191424
dims = 4
14201425
dims = get_test_dims(0, 8192, n=dims)
14211426
dims = [dim + (64 - (dim % 64)) for dim in dims]

0 commit comments

Comments
 (0)