Skip to content

Commit 801ecb7

Browse files
committed
Add comment
1 parent 002ce1a commit 801ecb7

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

cuda_bindings/cuda/bindings/nvml.pyx

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1593,6 +1593,9 @@ class FieldId(_FastEnum):
15931593

15941594
MAX = 289
15951595

1596+
# This value changed in CTK 13.3. We need to build a binary that works across
1597+
# all versions, so the only way to support this is to check the version at
1598+
# runtime and set the value accordingly.
15961599

15971600
if tuple(int(x) for x in system_get_nvml_version().split(".")) < (3, 13):
15981601
NVLINK_MAX_LINKS = 18

0 commit comments

Comments
 (0)