Xinya Zhang ef9451ac8d Move the build of AOTriton to base ROCM docker image. (#127012) ... Mitigates #126111 AOTrtion, as a Math library, takes long time to build. However, this library itself is not moving as fast as PyTorch itself and it is not cost-efficient to build it for every CI check. This PR moves the build of AOTriton from PyTorch to its base docker image, avoids duplicated and long build time. Pre-this-PR: * PyTorch base docker build job duration: 1.1-1.3h * PyTorch build job duration: 1.4-1.5hr (includes AOTriton build time of 1hr6min on a linux.2xlarge node) Post-this-PR: * PyTorch base docker build job duration: 1.3h (includes AOTriton build time of 20min on a linux.12xlarge node) * PyTorch build job duration: <20 min Co-authored-by: Jithun Nair <37884920+jithunnair-amd@users.noreply.github.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/127012 Approved by: https://github.com/jithunnair-amd, https://github.com/pruthvistony, https://github.com/huydhn		2024-06-03 20:35:22 +00:00
..
core	Remove more caffe2 files (#127511)	2024-05-31 11:26:27 +00:00
onnx/torch_ops
perfkernels
proto	[Caffe2] [Reland] Remove Caffe2 proto files (#127394)	2024-05-29 20:37:02 +00:00
serialize
utils	Remove more caffe2 files (#127511)	2024-05-31 11:26:27 +00:00
.clang-format
__init__.py
BUILD_MODE.bzl
CMakeLists.txt	Move the build of AOTriton to base ROCM docker image. (#127012)	2024-06-03 20:35:22 +00:00
unexported_symbols.lds
version_script.lds