Skip to content

Commit 5a91ad2

Browse files
test: skip linear8bitlt fullgraph compile test on torch < 2.8
fullgraph mode for torch.compile requires torch 2.8+, matching the existing guard in test_linear4bit_torch_compile. Previously masked by the torch < 2.4 skipif. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent 3353751 commit 5a91ad2

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

tests/test_linear8bitlt.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -257,6 +257,9 @@ def test_linear8bit_load_state_dict_raises_runtime_for_tied_weight():
257257
torch.__version__ < (2, 10) and sys.version_info >= (3, 14), reason="Not supported in Python 3.14 until torch 2.10"
258258
)
259259
def test_linear8bitlt_torch_compile(device, threshold, bias, fullgraph, mode):
260+
if fullgraph and torch.__version__ < (2, 8, 0, "dev"):
261+
pytest.skip("fullgraph mode requires torch 2.8 or higher")
262+
260263
if device == "cuda" and platform.system() == "Windows":
261264
pytest.skip("Triton is not officially supported on Windows")
262265

0 commit comments

Comments
 (0)