fix build system for thrust detection

leofang · Mar 7, 2024 · 1332f97 · 1332f97
1 parent 2fd0b81
commit 1332f97
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,