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
needs reproduction
Someone else needs to try reproducing the issue given the instructions. No action needed from user
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: