Merge pull request #8229 from leofang/thrust_build_fix

Fix build system for Thrust detection
cupy · Mar 8, 2024 · 7de16aa · 7de16aa
2 parents 2fd0b81 + 1332f97
commit 7de16aa
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/install/cupy_builder/_features.py b/install/cupy_builder/_features.py
@@ -390,9 +390,7 @@ def get_features(ctx: Context) -> Dict[str, Feature]:
             ('cupy.cuda.thrust', ['cupy/cuda/cupy_thrust.cu']),
         ],
         'include': [
-            'thrust/device_ptr.h',
-            'thrust/sequence.h',
-            'thrust/sort.h',
+            'thrust/version.h',
         ],
         'libraries': list(_cudart_static_libs),
         'check_method': build.check_thrust_version,