Merge pull request cupy#8530 from kmaehashi/fix-header-detection-conda

Fix cudart header detection for conda
chainer-ci · Aug 21, 2024 · 89d8fa0 · 89d8fa0
1 parent 87bd648
commit 89d8fa0
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/cupy/_core/core.pyx b/cupy/_core/core.pyx
@@ -2237,8 +2237,7 @@ cpdef tuple assemble_cupy_compiler_options(tuple options):
             wheel_dir_count = len(
                 _environment._get_include_dir_from_conda_or_wheel(
                     major, minor))
-            assert 0 <= wheel_dir_count <= 1
-            _headers_from_wheel_available = (wheel_dir_count == 1)
+            _headers_from_wheel_available = (0 < wheel_dir_count)
 
         if (_bundled_include is None and
                 _cuda_path is None and