leezu opened a new pull request #17521: cmake: don't build PTX and 3.5 arch if cuda arch detection fails URL: https://github.com/apache/incubator-mxnet/pull/17521 ## Description ## Currently, if `USE_CUDA=ON` but architecture detection fails due to building on a machine without GPU, we will build for `3.0;3.5;5.0;5.2;6.0;6.1;7.0;7.0+PTX;7.5;7.5+PTX`. Thereby we create too large object files, triggering https://github.com/apache/incubator-mxnet/issues/16852. Thus, this PR introduces patched `select_compute_arch.cmake`, such that the fallback arches are `3.0;5.0;5.2;6.0;6.1;7.0;7.5`, ie. dropping PTX and 3.5 so that #16852 is not triggered. We don't need 3.5, because code compiled for 3.0 also runs on 3.5 hardware and 3.5 hardware is 7 years old. If someone really cares about performance on 3.5, they can still manually set `MXNET_CUDA_ARCH=3.5`. Upstream bug https://gitlab.kitware.com/cmake/cmake/issues/20275 ### Changes ### - [X] cmake: don't build PTX and 3.5 arch if cuda arch detection fails
---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services
