Fix float128 on 32 bits (#1154)

* Fix float128 on 32 bits * support np.float96
automl · Jun 25, 2021 · 3cbdcc8 · 3cbdcc8
1 parent ffd6ed0
commit 3cbdcc8
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/autosklearn/automl.py b/autosklearn/automl.py
@@ -853,7 +853,14 @@ def subsample_if_too_large(
                 multiplier = 4
             elif X.dtype in (np.float64, np.float):
                 multiplier = 8
-            elif X.dtype == np.float128:
+            elif (
+                # In spite of the names, np.float96 and np.float128
+                # provide only as much precision as np.longdouble,
+                # that is, 80 bits on most x86 machines and 64 bits
+                # in standard Windows builds.
+                (hasattr(np, 'float128') and X.dtype == np.float128)
+                or (hasattr(np, 'float96') and X.dtype == np.float96)
+            ):
                 multiplier = 16
             else:
                 # Just assuming some value - very unlikely