diff --git a/setup.py b/setup.py index 1e08a5bd70cd3..994920ede349d 100644 --- a/setup.py +++ b/setup.py @@ -170,14 +170,17 @@ def configure(self, ext: CMakeExtension) -> None: if is_sccache_available(): cmake_args += [ + '-DCMAKE_C_COMPILER_LAUNCHER=sccache', '-DCMAKE_CXX_COMPILER_LAUNCHER=sccache', '-DCMAKE_CUDA_COMPILER_LAUNCHER=sccache', - '-DCMAKE_C_COMPILER_LAUNCHER=sccache', + '-DCMAKE_HIP_COMPILER_LAUNCHER=sccache', ] elif is_ccache_available(): cmake_args += [ + '-DCMAKE_C_COMPILER_LAUNCHER=ccache', '-DCMAKE_CXX_COMPILER_LAUNCHER=ccache', '-DCMAKE_CUDA_COMPILER_LAUNCHER=ccache', + '-DCMAKE_HIP_COMPILER_LAUNCHER=ccache', ] # Pass the python executable to cmake so it can find an exact