-
Notifications
You must be signed in to change notification settings - Fork 904
Description
Command:
python setup.py bdist_wheel
Error:
FAILED: /root/workdir/FlashMLA/build/temp.linux-aarch64-cpython-312/csrc/pybind.o
c++ -MMD -MF /root/workdir/FlashMLA/build/temp.linux-aarch64-cpython-312/csrc/pybind.o.d -fno-strict-overflow -Wsign-compare -DNDEBUG -g -O2 -Wall -fPIC -I/root/workdir/FlashMLA/csrc -I/root/workdir/FlashMLA/csrc/sm90 -I/root/workdir/FlashMLA/csrc/cutlass/include -I/root/workdir/FlashMLA/csrc/cutlass/tools/util/include -I/root/workdir/FlashMLA/.venv/lib/python3.12/site-packages/torch/include -I/root/workdir/FlashMLA/.venv/lib/python3.12/site-packages/torch/include/torch/csrc/api/include -I/usr/local/cuda-13.0/include -I/root/workdir/FlashMLA/.venv/include -I/usr/include/python3.12 -c -c /root/workdir/FlashMLA/csrc/pybind.cpp -o /root/workdir/FlashMLA/build/temp.linux-aarch64-cpython-312/csrc/pybind.o -O3 -std=c++17 -DNDEBUG -Wno-deprecated-declarations -DTORCH_API_INCLUDE_EXTENSION_H -DTORCH_EXTENSION_NAME=cuda
In file included from /root/workdir/FlashMLA/csrc/pybind.cpp:11:
/root/workdir/FlashMLA/csrc/cutlass/include/cutlass/fast_math.h:42:10: fatal error: cuda/std/utility: No such file or directory
42 | #include <cuda/std/utility>