pytorch
bb89a9e4 - Skipped CUDA Flags if C++ Extension Name includes "arch" Substring (#111211)

Commit
1 year ago
Skipped CUDA Flags if C++ Extension Name includes "arch" Substring (#111211) The CUDA architecture flags from TORCH_CUDA_ARCH_LIST will be skipped if the TORCH_EXTENSION_NAME includes the substring "arch". A C++ Extension should be allowed to have any name. I just manually skip the TORCH_EXTENSION_NAME flag when checking if one of the flags is "arch". There is probably a better fix, but I'll leave this to experts. Pull Request resolved: https://github.com/pytorch/pytorch/pull/111211 Approved by: https://github.com/ezyang
Author
Committer
Parents
Loading