fix(ci): handle conda cuda target layout

Author: abetlen

.github/workflows/build-wheels-cuda.yaml

@@ -109,11 +109,24 @@ jobs:
           $env:CUDA_HOME = $env:CONDA_PREFIX
           $env:CUDA_TOOLKIT_ROOT_DIR = $env:CONDA_PREFIX
           if ($IsLinux) {
-            $env:LD_LIBRARY_PATH = $env:CONDA_PREFIX + '/lib:' + $env:LD_LIBRARY_PATH
+            $cudaTargetRoot = Join-Path $env:CONDA_PREFIX 'targets/x86_64-linux'
+            if (Test-Path (Join-Path $cudaTargetRoot 'include/cuda_runtime.h')) {
+              $env:CUDAToolkit_ROOT = $cudaTargetRoot
+              $env:CUDA_TOOLKIT_ROOT_DIR = $cudaTargetRoot
+              $env:CMAKE_ARGS = "-DCUDAToolkit_ROOT=$cudaTargetRoot -DCUDA_TOOLKIT_ROOT_DIR=$cudaTargetRoot"
+              $env:CPATH = "$cudaTargetRoot/include:$env:CPATH"
+              $env:CPLUS_INCLUDE_PATH = "$cudaTargetRoot/include:$env:CPLUS_INCLUDE_PATH"
+              $env:LIBRARY_PATH = "$cudaTargetRoot/lib:$env:LIBRARY_PATH"
+              $env:LD_LIBRARY_PATH = "$cudaTargetRoot/lib:$env:CONDA_PREFIX/lib:$env:LD_LIBRARY_PATH"
+            } else {
+              $env:LD_LIBRARY_PATH = $env:CONDA_PREFIX + '/lib:' + $env:LD_LIBRARY_PATH
+              $env:CMAKE_ARGS = ''
+            }
+          } else {
+            $env:CMAKE_ARGS = ''
           }
           $env:VERBOSE = '1'
-          $env:CMAKE_ARGS = '-DGGML_CUDA=on -DCMAKE_CUDA_ARCHITECTURES=all'
-          $env:CMAKE_ARGS = "-DGGML_CUDA_FORCE_MMQ=ON $env:CMAKE_ARGS"
+          $env:CMAKE_ARGS = "-DGGML_CUDA_FORCE_MMQ=ON -DGGML_CUDA=on -DCMAKE_CUDA_ARCHITECTURES=all $env:CMAKE_ARGS"
           # if ($env:AVXVER -eq 'AVX') {
           $env:CMAKE_ARGS = $env:CMAKE_ARGS + ' -DGGML_AVX2=off -DGGML_FMA=off -DGGML_F16C=off'
           # }
@@ -128,6 +141,7 @@ jobs:
           Write-Output "CUDA_VERSION=$cudaVersion" >> $env:GITHUB_ENV
 
       - uses: softprops/action-gh-release@v2
+        if: startsWith(github.ref, 'refs/tags/')
         with:
           files: dist/*
           # Set tag_name to <tag>-cu<cuda_version>