dev(hansbug): disable torch.compile on windows, install libomp for macos

opendilab · Apr 16, 2023 · 1fef868 · 1fef868
1 parent fc627ee
commit 1fef868
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -85,6 +85,7 @@ jobs:
         shell: bash
         run: |
           brew install tree cloc wget curl make zip graphviz
+          brew install libomp
           dot -V
       - name: Set up python ${{ matrix.python-version }}
         uses: actions/setup-python@v4

diff --git a/test/tree/integration/test_torch.py b/test/tree/integration/test_torch.py
@@ -1,7 +1,7 @@
 from unittest import skipUnless
 
 import pytest
-from hbutils.testing import vpip
+from hbutils.testing import vpip, OS
 
 from treevalue import FastTreeValue, register_for_torch
 
@@ -62,7 +62,7 @@ class MyTreeValueX(FastTreeValue):
         with pytest.warns(UserWarning):
             register_for_torch(MyTreeValueX)
 
-    @skipUnless(vpip('torch') >= '2.0.0', 'Torch 2 required')
+    @skipUnless(vpip('torch') >= '2.0.0' and not OS.windows, 'Torch 2 on non-windows platform required')
     def test_torch_compile(self):
         @torch.compile
         def foo(x, y, t):