expose TORCH_CUDA_ARCH_LIST#147

Open

ErwannMillon wants to merge 2 commits intothu-ml:mainfrom

ErwannMillon:main

Commits on Mar 24, 2025

expose TORCH_CUDA_ARCH_LIST
ErwannMillon
committed

Commits on Mar 25, 2025

override automatically detected cuda architectures if TORCH_CUDA_ARCH_LIST is passed
ErwannMillon
committed