pytorch/cmake
Jithun Nair 851b9732aa Download pre-compiled AOTriton from GitHub unless AOTRITON_INSTALL_FROM_SOURCE=1 is set (#136603)
PyTorch community members have reported issues with building PyTorch from source for ROCm in an environment that doesn't have aotriton pre-installed, because aotriton is only installed in the [CI](a8ed873ba2/.ci/docker/manywheel/Dockerfile (L197)) docker images. Building aotriton from source can take ~45 minutes.

This PR fixes the issue by downloading the aotriton tarball in such scenarios, *unless the user explicitly wants to build aotriton from source using the AOTRITON_INSTALL_FROM_SOURCE=1 env var*

Pull Request resolved: https://github.com/pytorch/pytorch/pull/136603
Approved by: https://github.com/atalman

Co-authored-by: Xinya Zhang <Xinya.Zhang@amd.com>
2024-09-26 18:05:51 +00:00
..
External Download pre-compiled AOTriton from GitHub unless AOTRITON_INSTALL_FROM_SOURCE=1 is set (#136603) 2024-09-26 18:05:51 +00:00
Modules Extending the Pytorch vec backend for SVE (ARM) (#119571) 2024-09-18 18:59:10 +00:00
Modules_CUDA_fix
public Fix hardcoded ROCm paths in Caffe2Targets.cmake (#136283) 2024-09-26 00:34:43 +00:00
Allowlist.cmake
BuildVariables.cmake
Caffe2Config.cmake.in
CheckAbi.cmake
cmake_uninstall.cmake.in
Codegen.cmake Extending the Pytorch vec backend for SVE (ARM) (#119571) 2024-09-18 18:59:10 +00:00
DebugHelper.cmake
Dependencies.cmake Fix hardcoded ROCm paths in Caffe2Targets.cmake (#136283) 2024-09-26 00:34:43 +00:00
FlatBuffers.cmake
GoogleTestPatch.cmake
IncludeSource.cpp.in
iOS.cmake
Metal.cmake
MiscCheck.cmake Clarify that libtorch API is C++17 compatible (#136471) 2024-09-24 02:03:33 +00:00
prioritized_text.txt
ProtoBuf.cmake
ProtoBufPatch.cmake
Summary.cmake Revert "Adds support for accelerated sorting with x86-simd-sort (#127936)" 2024-09-23 14:52:23 +00:00
TorchConfig.cmake.in [Reland2] Update NVTX to NVTX3 (#109843) 2024-08-20 16:33:26 +00:00
TorchConfigVersion.cmake.in
VulkanCodegen.cmake
VulkanDependencies.cmake