提交 63ded6e8 编写于 作者: HansBug's avatar HansBug 😆

dev(hansbug): linux only for torch.compile

上级 1fef8687
...@@ -85,7 +85,6 @@ jobs: ...@@ -85,7 +85,6 @@ jobs:
shell: bash shell: bash
run: | run: |
brew install tree cloc wget curl make zip graphviz brew install tree cloc wget curl make zip graphviz
brew install libomp
dot -V dot -V
- name: Set up python ${{ matrix.python-version }} - name: Set up python ${{ matrix.python-version }}
uses: actions/setup-python@v4 uses: actions/setup-python@v4
......
...@@ -62,7 +62,7 @@ class TestTreeIntegrationTorch: ...@@ -62,7 +62,7 @@ class TestTreeIntegrationTorch:
with pytest.warns(UserWarning): with pytest.warns(UserWarning):
register_for_torch(MyTreeValueX) register_for_torch(MyTreeValueX)
@skipUnless(vpip('torch') >= '2.0.0' and not OS.windows, 'Torch 2 on non-windows platform required') @skipUnless(vpip('torch') >= '2.0.0' and OS.linux, 'Torch 2 on linux platform required')
def test_torch_compile(self): def test_torch_compile(self):
@torch.compile @torch.compile
def foo(x, y, t): def foo(x, y, t):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册