remove no b units

rapidsai · Sep 6, 2024 · 4f3db58 · 4f3db58
1 parent 0e6eee0
commit 4f3db58
Showing 1 changed file with 1 addition and 12 deletions.
diff --git a/python/rmm/rmm/_lib/helper.pyx b/python/rmm/rmm/_lib/helper.pyx
@@ -25,26 +25,15 @@ cdef dict BYTE_SIZES = {
     'gb': 1000**3,
     'tb': 1000**4,
     'pb': 1000**5,
-    'k': 1000,
-    'm': 1000**2,
-    'g': 1000**3,
-    't': 1000**4,
-    'p': 1000**5,
     'kib': 1024,
     'mib': 1024**2,
     'gib': 1024**3,
     'tib': 1024**4,
     'pib': 1024**5,
-    'ki': 1024,
-    'mi': 1024**2,
-    'gi': 1024**3,
-    'ti': 1024**4,
-    'pi': 1024**5,
 }
 
 
-pattern = re.compile(r"^([0-9]+(?:\.[0-9]*)?)[\t ]*((?i:(?:[kmgtp]i?)?b)?)$")
-
+pattern = re.compile(r"^([0-9]+(?:\.[0-9]*)?)[\t ]*((?i:(?:[kmgtp]i?)?b))?$")
 
 cdef object parse_bytes(object s):
     """ Parse byte string to numbers