Test torch==1.12.1 with OMP_NUM_THREADS=3 #300

neuml · Aug 5, 2022 · 9b28b7a · 9b28b7a
1 parent dd3191e
commit 9b28b7a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -28,6 +28,7 @@ jobs:
         run: |
           brew install libomp
           pip install annoy==1.16.3
+        env: OMP_NUM_THREADS: 3
         if: matrix.os == 'macos-latest'
 
       - name: Install dependencies - Windows
@@ -38,7 +39,7 @@ jobs:
         run: |
           pip install -U pip
           pip install -U wheel coverage coveralls
-          pip install .[all] torch==1.11.0
+          pip install .[all]
           python -c "import nltk; nltk.download('punkt')"
           python --version
           make data coverage