pytorch
59a3759d - Update cpp_extension.py (#101285)

Commit
1 year ago
Update cpp_extension.py (#101285) When we need to link extra libs, we should notice that 64-bit CUDA may be installed in "lib", not in "lib64". <!-- copilot:summary --> ### <samp>🤖 Generated by Copilot at 05c1ca6</samp> Improve CUDA compatibility in `torch.utils.cpp_extension` by checking for `lib64` or `lib` directory. Pull Request resolved: https://github.com/pytorch/pytorch/pull/101285 Approved by: https://github.com/ezyang, https://github.com/malfet
Author
Committer
Parents
Loading