Skip to content

Commit

Permalink
Add CUDA 12.4 to supported PTX versions (#15247)
Browse files Browse the repository at this point in the history
This PR updates the mapping from PTX version to toolkit versions to cover CUDA 12.4.

Authors:
  - https://github.com/brandon-b-miller

Approvers:
  - Bradley Dice (https://github.com/bdice)

URL: #15247
  • Loading branch information
brandon-b-miller authored Mar 7, 2024
1 parent 753bf3e commit 188d7cb
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions python/cudf/cudf/utils/_numba.py
Original file line number Diff line number Diff line change
Expand Up @@ -181,6 +181,7 @@ def _get_cuda_version_from_ptx_file(path):
"8.1": (12, 1),
"8.2": (12, 2),
"8.3": (12, 3),
"8.4": (12, 4),
}

cuda_ver = ver_map.get(version)
Expand Down

0 comments on commit 188d7cb

Please sign in to comment.