Update python/kvikio/kvikio/cufile_driver.py

Co-authored-by: jakirkham <[email protected]>
rapidsai · Dec 5, 2024 · e4e9dfa · e4e9dfa
1 parent c231396
commit e4e9dfa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/kvikio/kvikio/cufile_driver.py b/python/kvikio/kvikio/cufile_driver.py
@@ -26,8 +26,8 @@ def libcufile_version() -> Tuple[int, int]:
     The version as a tuple (MAJOR, MINOR).
     """
     v = cufile_driver.libcufile_version()
-    major = v // 1000
-    minor = (v % 1000) // 10
+    major, minor = divmod(v, 1000)
+    minor //= 10
     return (major, minor)