setup.py · d5ca4495c0c068bc617102841b9322d378fac8ea · OpenDAS / DeepEP

Make `TORCH_CUDA_ARCH_LIST` as an environment variable (#167) · d5ca4495

guyueh1 authored May 18, 2025



* Add 10.0 to TORCH_CUDA_ARCH_LIST
Signed-off-by: Guyue Huang <guyueh@nvidia.com>

* Revert csrc/CMakeLists change; in setup.py make TORCH_CUDA_ARCH_LIST configurable
Signed-off-by: Guyue Huang <guyueh@nvidia.com>

---------
Signed-off-by: Guyue Huang <guyueh@nvidia.com>

d5ca4495

setup.py 2.37 KB

Replace setup.py