From c132d189fdd00866545b86e07244c7d0cb8095a8 Mon Sep 17 00:00:00 2001 From: lcskrishna Date: Thu, 9 Jan 2025 06:46:42 +0000 Subject: [PATCH] add thrust macros for rocm6.3 --- install/cupy_builder/install_build.py | 1 + 1 file changed, 1 insertion(+) 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()