add thrust macros for rocm6.3

ROCm · Jan 9, 2025 · c132d18 · c132d18
1 parent 25b5b33
commit c132d18
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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()