Cannot install CuPy from source on ROCm 6.0.2

[fxzjshm]

Description

Firstly, building CuPy on ROCm requires additional CFLAGS:

export CFLAGS="-I/opt/rocm/include/hipblas -I/opt/rocm/include/hipsparse -I/opt/rocm/include/hipfft -I/opt/rocm/include/rocsolver -I/opt/rocm/include/rccl"

But pip install cupy still fails after several minutes:

error: Command "x86_64-linux-gnu-gcc -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -O2 -Wall -g -fstack-protector-strong -Wformat -Werror=format-security -g -fwrapv -O2 -I/opt/rocm/include/hipblas -I/opt/rocm/include/hipsparse -I/opt/rocm/include/hipfft -I/opt/rocm/include/rocsolver -I/opt/rocm/include/rccl -Wdate-time -D_FORTIFY_SOURCE=2 -fPIC -D_FORCE_INLINES=1 -DCUPY_USE_HIP=1 -D__HIP_PLATFORM_AMD__=1 -D__HIP_PLATFORM_HCC__=1 -DCUPY_CACHE_KEY=d4770f4cb60796d63570de639682672dffe05289 -DCUPY_CUB_VERSION_CODE=300000 -I/opt/rocm/hipfft/include -I/opt/rocm/include/hipcub -I/tmp/pip-install-0x7cot68/cupy_f9bce70497514fa68cfe3560b85351d4/cupy/_core/include -I/opt/rocm/include -I/opt/rocm/include/hip -I/opt/rocm/include/rocrand -I/opt/rocm/include/hiprand -I/opt/rocm/include/roctracer -I/usr/include/python3.10 -c cupy_backends/cuda/api/runtime.cpp -o build/temp.linux-x86_64-3.10/cupy_backends/cuda/api/runtime.o -std=c++11" failed with exit status 1

(It's strange that cupy_backends/cuda/api/runtime.cpp is compiled with x86_64-linux-gnu-gcc)

To Reproduce

export CFLAGS="-I/opt/rocm/include/hipblas -I/opt/rocm/include/hipsparse -I/opt/rocm/include/hipfft -I/opt/rocm/include/rocsolver"
export CUPY_INSTALL_USE_HIP=1
export ROCM_HOME=/opt/rocm
export HCC_AMDGPU_TARGET=gfx906
sudo -E pip3 install cupy

Installation

Source (pip install cupy)

Environment

Not available

Additional Information

pip-cupy.log

[remy-abergel]

I am facing more or less the same problem but my compilation attempt ends up with a different error.

To reproduce

export HCC_AMDGPU_TARGET=gfx803
export CUPY_INSTALL_USE_HIP=1
export ROCM_HOME=/opt/rocm-6.0.2
export CFLAGS="-I/opt/rocm-6.0.2/include/hipblas -I/opt/rocm-6.0.2/include/hipsparse -I/opt/rocm-6.0.2/include/hipfft -I/opt/rocm-6.0.2/include/rocsolver -I/opt/rocm-6.0.2/include/rccl"
pip install cupy --log pip-cupy.log

(notice that I added the include path for rccl into CFLAGS)

Additional Information

pip-cupy.log

Thank you!

[jaghili]

I added the missing include dirs ininstall_build.py and run python setup.py install but I still get compilation errors:

errors.log
output.log