diff --git a/install/cupy_builder/install_build.py b/install/cupy_builder/install_build.py index 879a2089be4..fb280b58beb 100644 --- a/install/cupy_builder/install_build.py +++ b/install/cupy_builder/install_build.py @@ -159,6 +159,7 @@ def get_compiler_setting(ctx: Context, use_hip): if use_hip: extra_compile_args.append('-std=c++11') + define_macros.append(('THRUST_DEVICE_SYSTEM', 'THRUST_DEVICE_SYSTEM_HIP')) if PLATFORM_WIN32: nvtx_path = _environment.get_nvtx_path()