fix typos in workflows

OpenBMB · Feb 16, 2024 · c1a1ef3 · c1a1ef3
1 parent d55f36d
commit c1a1ef3
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 12 deletions.
diff --git a/.github/workflows/build_wheels_cuda_linux.yml b/.github/workflows/build_wheels_cuda_linux.yml
@@ -94,11 +94,12 @@ jobs:
 
       - name: Build Wheel
         run: |
-          export CUDA_PATH=$CONDA_PREFIX
-          export CUDA_HOME=$CONDA_PREFIX
+          # For some reason $CONDA_PREFIX is empty.
+          export CUDA_HOME=/usr/share/miniconda
+          export CUDA_PATH=/usr/share/miniconda
           export LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:${CONDA_PREFIX}/lib"
 
-          export TORCH_CUDA_ARCH_LIST = '6.0 6.1 7.0 7.5 8.0 8.6 8.9 9.0+PTX'
+          export TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6 8.9 9.0+PTX"
 
           echo "CUDA_PATH:"
           echo $CUDA_PATH

diff --git a/.github/workflows/build_wheels_pypi_linux.yml b/.github/workflows/build_wheels_pypi_linux.yml
@@ -93,11 +93,12 @@ jobs:
 
       - name: Build Wheel
         run: |
-          export CUDA_PATH=$CONDA_PREFIX
-          export CUDA_HOME=$CONDA_PREFIX
+          # For some reason $CONDA_PREFIX is empty.
+          export CUDA_HOME=/usr/share/miniconda
+          export CUDA_PATH=/usr/share/miniconda
           export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:${CONDA_PREFIX}/lib"
 
-          export TORCH_CUDA_ARCH_LIST='6.0 6.1 7.0 7.5 8.0 8.6 8.9 9.0+PTX'
+          export TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6 8.9 9.0+PTX"
 
           export PYPI_RELEASE="1"
 

diff --git a/.github/workflows/test_quality.yml b/.github/workflows/test_quality.yml
@@ -97,11 +97,7 @@ jobs:
           conda install cuda-toolkit -c "nvidia/label/cuda-12.1.0"
 
           # Refer to https://pytorch.org/get-started/locally/
-          if [[ "${{ runner.os }}" == "Linux" ]]; then
-            python -m pip install torch
-          else
-            python -m pip install torch --index-url https://download.pytorch.org/whl/cu121
-          fi
+          python -m pip install torch
           
           python -m pip install --upgrade build setuptools wheel numpy
 
@@ -121,7 +117,7 @@ jobs:
           echo "CUDA_PATH:"
           echo $CUDA_PATH
 
-          TORCH_CUDA_ARCH_LIST="7.5 8.0 8.6+PTX" DISABLE_QIGEN=1 pip install -vvv .[quality]
+          pip install -vvv .[quality]
       
       - name: Check style with ruff
         run: |