pytorch

8940a4e6 - Pull upstream select_compute_arch from cmake for Ampere (#41133)

Pull upstream select_compute_arch from cmake for Ampere (#41133) Summary: This pulls the following merge requests from CMake upstream: - https://gitlab.kitware.com/cmake/cmake/-/merge_requests/4979 - https://gitlab.kitware.com/cmake/cmake/-/merge_requests/4991 The above two merge requests improve the Ampere build: - If `TORCH_CUDA_ARCH_LIST` is not set, it can now automatically pickup 8.0 as its part of its default value - If `TORCH_CUDA_ARCH_LIST=Ampere`, it no longer fails with `Unknown CUDA Architecture Name Ampere in CUDA_SELECT_NVCC_ARCH_FLAGS` Codes related to architecture < 3.5 are manually removed because PyTorch no longer supports it. cc: ngimel ptrblck Pull Request resolved: https://github.com/pytorch/pytorch/pull/41133 Reviewed By: malfet Differential Revision: D22540547 Pulled By: ezyang fbshipit-source-id: 6e040f4054ef04f18ebb7513497905886a375632