ROCm: fatal error: aotriton/flash.h: No such file or directory
when building with USE_ROCM=1
#125230
Labels
bug
module: rocm
AMD GPU support for Pytorch
triaged
This issue has been looked at a team member, and triaged and prioritized into an appropriate module
🐛 Describe the bug
I am running in
rocm/dev-ubuntu-22.04:6.1
.Building PyTorch 2.3 (https://github.com/pytorch/pytorch/tree/release/2.3) with
and getting
around the end of the build.
It appears that the file
aotriton/flash.h
is not in PyTorch codebase. It is here though: https://github.com/ROCm/aotriton (which says the install should be smooth through pytorch https://github.com/ROCm/aotriton?tab=readme-ov-file#pytorch-consumption, so I am confused).Is something wrong in https://github.com/pytorch/pytorch/blob/main/cmake/External/aotriton.cmake? Maybe need an
include_directories(build/aotriton/src/include)
?Apparently during the build both
./build/aotriton/src/include/aotriton/flash.h
and./torch/include/aotriton/flash.h
are generated, but not used.Thank you!
Versions
cc @jeffdaily @sunway513 @jithunnair-amd @pruthvistony @ROCmSupport @dllehr-amd @jataylo @hongxiayang
The text was updated successfully, but these errors were encountered: