Skip to content

Commit

Permalink
Temporarily disable some inductor tests on windows
Browse files Browse the repository at this point in the history
These tests require that the `cl` compiler is available and this
may take some setup.
  • Loading branch information
tfogal committed Oct 17, 2024
1 parent 953a914 commit 9fc1c78
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 2 deletions.
5 changes: 4 additions & 1 deletion thunder/tests/test_autocast.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
import itertools
import platform

import pytest
import torch
Expand Down Expand Up @@ -140,7 +141,9 @@ def func(a, b):
assert output.dtype == (torch.float16 if torch_device.type == "cuda" else torch.bfloat16)


@pytest.mark.skipif(not is_inductor_supported(), reason="inductor unsupported")
# Disabling on windows temporarily, until our windows runners source the
# appropriate visual studio config.
@pytest.mark.skipif(not is_inductor_supported() or platform.system() == "Windows", reason="inductor unsupported")
def test_torch_compile_autocast():
"""Checks if our autocast decorator plays well with ``torch.compile``"""

Expand Down
5 changes: 4 additions & 1 deletion thunder/tests/test_torch_compile_executor.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
import platform
import pytest
import torch
from torch._dynamo import is_inductor_supported
Expand All @@ -15,7 +16,9 @@ def test_supported_ops_are_in_pytorch_executor():
assert supported_ops - pytorch_ex.implmap.keys() == set()


@pytest.mark.skipif(not is_inductor_supported(), reason="inductor unsupported")
# Disabling on windows temporarily, until our windows runners source the
# appropriate visual studio config.
@pytest.mark.skipif(not is_inductor_supported() or platform.system() == "Windows", reason="inductor unsupported")
def test_torch_compile_litgpt():
from litgpt.model import GPT

Expand Down

0 comments on commit 9fc1c78

Please sign in to comment.