Skip to content

Commit

Permalink
Merge pull request cupy#8505 from kmaehashi/fix-8466-ver
Browse files Browse the repository at this point in the history
Fix CUDA version condition to use headers from wheel
  • Loading branch information
kmaehashi authored and chainer-ci committed Aug 17, 2024
1 parent a1c4326 commit 2282640
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion cupy/_environment.py
Original file line number Diff line number Diff line change
Expand Up @@ -454,7 +454,7 @@ def _preload_warning(lib, exc):
def _get_include_dir_from_wheel(major: int, minor: int) -> List[str]:
# FP16 headers from CUDA 12.2+ depends on headers from CUDA Runtime.
# See https://github.com/cupy/cupy/issues/8466.
if major < 12 or (major == 12 and minor < 3):
if major < 12 or (major == 12 and minor < 2):
return []

config = get_preload_config()
Expand Down

0 comments on commit 2282640

Please sign in to comment.