fix: restore pytorch DLL replacement

metastable-studio · Jan 4, 2025 · d18ca43 · d18ca43
1 parent f33804c
commit d18ca43
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -23,15 +23,20 @@ jobs:
     steps:
       - name: Prepare ZLUDA
         run: |
+          mkdir -p python/Lib/site-packages/torch/lib
           curl -L -o zluda.zip ${{ matrix.build.zluda_url }}
           unzip zluda.zip
           cp zluda/cublas.dll zluda/cublas64_11.dll
           cp zluda/cusparse.dll zluda/cusparse64_11.dll
           cp zluda/nvrtc.dll zluda/nvrtc64_112_0.dll
+          cp zluda/cublas.dll python/Lib/site-packages/torch/lib/cublas64_11.dll
+          cp zluda/cusparse.dll python/Lib/site-packages/torch/lib/cusparse64_11.dll
+          cp zluda/nvrtc.dll python/Lib/site-packages/torch/lib/nvrtc64_112_0.dll
+          cp -r zluda python/zluda
 
       - name: Build artifact
         run: |
-          cd zluda
+          cd python
           tar -cf - * | brotli -6 > /tmp/${{ env.artifact_name }}
 
       - name: Upload artifact