Skip to content

Commit

Permalink
Attempt reenabling cuDSS, cuSPARSELt and CUPTI on Windows
Browse files Browse the repository at this point in the history
Fixes #319
Fixes #320
Fixes #321
  • Loading branch information
mgorny committed Jan 19, 2025
1 parent 39b84c3 commit c3b2a1f
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 12 deletions.
6 changes: 0 additions & 6 deletions recipe/bld.bat
Original file line number Diff line number Diff line change
Expand Up @@ -26,12 +26,6 @@ if "%blas_impl%" == "generic" (
SET "USE_NUMA=0"
SET "USE_ITT=0"

@REM KINETO seems to require CUPTI and will look quite hard for it.
@REM CUPTI seems to cause trouble when users install a version of
@REM cudatoolkit different than the one specified at compile time.
@REM https://github.com/conda-forge/pytorch-cpu-feedstock/issues/135
set "USE_KINETO=OFF"

if "%PKG_NAME%" == "pytorch" (
set "PIP_ACTION=install"
:: We build libtorch for a specific python version.
Expand Down
12 changes: 6 additions & 6 deletions recipe/meta.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -126,14 +126,14 @@ requirements:
{% if cuda_compiler_version != "None" %}
- cuda-driver-dev # [linux]
- cuda-cudart-dev
- cuda-cupti-dev # [linux]
- cuda-cupti-dev
- cuda-nvrtc-dev
- cuda-nvtx-dev
- cuda-nvml-dev
- cuda-profiler-api
- cusparselt # [linux]
- cusparselt
- libcublas-dev
- libcudss-dev # [linux]
- libcudss-dev
- libcufile-dev # [linux]
- libcufft-dev
- libcurand-dev
Expand Down Expand Up @@ -266,14 +266,14 @@ outputs:
{% if cuda_compiler_version != "None" %}
- cuda-driver-dev # [linux]
- cuda-cudart-dev
- cuda-cupti-dev # [linux]
- cuda-cupti-dev
- cuda-nvrtc-dev
- cuda-nvtx-dev
- cuda-nvml-dev
- cuda-profiler-api
- cusparselt # [linux]
- cusparselt
- libcublas-dev
- libcudss-dev # [linux]
- libcudss-dev
- libcufile-dev # [linux]
- libcufft-dev
- libcurand-dev
Expand Down

0 comments on commit c3b2a1f

Please sign in to comment.