File tree Expand file tree Collapse file tree 2 files changed +8
-2
lines changed Expand file tree Collapse file tree 2 files changed +8
-2
lines changed Original file line number Diff line number Diff line change 1
1
import itertools
2
+ import platform
2
3
3
4
import pytest
4
5
import torch
@@ -140,7 +141,9 @@ def func(a, b):
140
141
assert output .dtype == (torch .float16 if torch_device .type == "cuda" else torch .bfloat16 )
141
142
142
143
143
- @pytest .mark .skipif (not is_inductor_supported (), reason = "inductor unsupported" )
144
+ # Disabling on windows temporarily, until our windows runners source the
145
+ # appropriate visual studio config.
146
+ @pytest .mark .skipif (not is_inductor_supported () or platform .system () == "Windows" , reason = "inductor unsupported" )
144
147
def test_torch_compile_autocast ():
145
148
"""Checks if our autocast decorator plays well with ``torch.compile``"""
146
149
Original file line number Diff line number Diff line change
1
+ import platform
1
2
import pytest
2
3
import torch
3
4
from torch ._dynamo import is_inductor_supported
@@ -15,7 +16,9 @@ def test_supported_ops_are_in_pytorch_executor():
15
16
assert supported_ops - pytorch_ex .implmap .keys () == set ()
16
17
17
18
18
- @pytest .mark .skipif (not is_inductor_supported (), reason = "inductor unsupported" )
19
+ # Disabling on windows temporarily, until our windows runners source the
20
+ # appropriate visual studio config.
21
+ @pytest .mark .skipif (not is_inductor_supported () or platform .system () == "Windows" , reason = "inductor unsupported" )
19
22
def test_torch_compile_litgpt ():
20
23
from litgpt .model import GPT
21
24
You can’t perform that action at this time.
0 commit comments