Handle float128 generically

hcho3 · May 23, 2024 · b80baf4 · b80baf4
1 parent 95ba099
commit b80baf4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python-package/xgboost/collective.py b/python-package/xgboost/collective.py
@@ -184,8 +184,10 @@ def _map_dtype(dtype: np.dtype) -> int:
         np.dtype("uint32"): 10,
         np.dtype("uint64"): 11,
     }
-    if platform.system() != "Windows":
+    try:
         dtype_map.update({np.dtype("float128"): 3})
+    except TypeError:  # float128 doesn't exist on the system
+        pass
 
     if dtype not in dtype_map:
         raise TypeError(f"data type {dtype} is not supported on the current platform.")