Support for all primitive types from array. (#7003)

* Change C API name.
* Test for all primitive types from array.
* Add native support for CPU 128 float.
* Convert boolean and float16 in Python.

* Fix dask version for now.

tests/ci_build/conda_env/macos_cpu_test.yml

@@ -13,8 +13,8 @@ dependencies:
 - scikit-learn
 - pandas
 - matplotlib
-- dask
-- distributed
+- dask=2021.05.0
+- distributed=2021.05.0
 - graphviz
 - python-graphviz
 - hypothesis

tests/python-gpu/test_gpu_prediction.py

@@ -204,6 +204,7 @@ class TestGPUPredict:
         cpu_predt = reg.predict(X)
         np.testing.assert_allclose(gpu_predt, cpu_predt, atol=1e-6)
 
+    @pytest.mark.skipif(**tm.no_cupy())
     @pytest.mark.skipif(**tm.no_cudf())
     def test_inplace_predict_cudf(self):
         import cupy as cp
@@ -332,6 +333,7 @@ class TestGPUPredict:
         rmse = mean_squared_error(y_true=y, y_pred=pred, squared=False)
         np.testing.assert_almost_equal(rmse, eval_history['train']['rmse'][-1], decimal=5)
 
+    @pytest.mark.skipif(**tm.no_cupy())
     @pytest.mark.parametrize("n_classes", [2, 3])
     def test_predict_dart(self, n_classes):
         from sklearn.datasets import make_classification
@@ -378,3 +380,59 @@ class TestGPUPredict:
 
         copied = cp.array(copied)
         cp.testing.assert_allclose(inplace, copied, atol=1e-6)
+
+    @pytest.mark.skipif(**tm.no_cupy())
+    def test_dtypes(self):
+        import cupy as cp
+        rows = 1000
+        cols = 10
+        rng = cp.random.RandomState(1994)
+        orig = rng.randint(low=0, high=127, size=rows * cols).reshape(
+            rows, cols
+        )
+        y = rng.randint(low=0, high=127, size=rows)
+        dtrain = xgb.DMatrix(orig, label=y)
+        booster = xgb.train({"tree_method": "gpu_hist"}, dtrain)
+
+        predt_orig = booster.inplace_predict(orig)
+        # all primitive types in numpy
+        for dtype in [
+            cp.signedinteger,
+            cp.byte,
+            cp.short,
+            cp.intc,
+            cp.int_,
+            cp.longlong,
+            cp.unsignedinteger,
+            cp.ubyte,
+            cp.ushort,
+            cp.uintc,
+            cp.uint,
+            cp.ulonglong,
+            cp.floating,
+            cp.half,
+            cp.single,
+            cp.double,
+        ]:
+            X = cp.array(orig, dtype=dtype)
+            predt = booster.inplace_predict(X)
+            cp.testing.assert_allclose(predt, predt_orig)
+
+        # boolean
+        orig = cp.random.binomial(1, 0.5, size=rows * cols).reshape(
+            rows, cols
+        )
+        predt_orig = booster.inplace_predict(orig)
+        for dtype in [cp.bool8, cp.bool_]:
+            X = cp.array(orig, dtype=dtype)
+            predt = booster.inplace_predict(X)
+            cp.testing.assert_allclose(predt, predt_orig)
+
+        # unsupported types
+        for dtype in [
+            cp.complex64,
+            cp.complex128,
+        ]:
+            X = cp.array(orig, dtype=dtype)
+            with pytest.raises(ValueError):
+                booster.inplace_predict(X)

tests/python/test_predict.py

@@ -237,3 +237,51 @@ class TestInplacePredict:
         dtrain = xgb.DMatrix(self.X, self.y, base_margin=base_margin)
         from_dmatrix = booster.predict(dtrain)
         np.testing.assert_allclose(from_dmatrix, from_inplace)
+
+    def test_dtypes(self):
+        orig = self.rng.randint(low=0, high=127, size=self.rows * self.cols).reshape(
+            self.rows, self.cols
+        )
+        predt_orig = self.booster.inplace_predict(orig)
+        # all primitive types in numpy
+        for dtype in [
+            np.signedinteger,
+            np.byte,
+            np.short,
+            np.intc,
+            np.int_,
+            np.longlong,
+            np.unsignedinteger,
+            np.ubyte,
+            np.ushort,
+            np.uintc,
+            np.uint,
+            np.ulonglong,
+            np.floating,
+            np.half,
+            np.single,
+            np.double,
+        ]:
+            X = np.array(orig, dtype=dtype)
+            predt = self.booster.inplace_predict(X)
+            np.testing.assert_allclose(predt, predt_orig)
+
+        # boolean
+        orig = self.rng.binomial(1, 0.5, size=self.rows * self.cols).reshape(
+            self.rows, self.cols
+        )
+        predt_orig = self.booster.inplace_predict(orig)
+        for dtype in [np.bool8, np.bool_]:
+            X = np.array(orig, dtype=dtype)
+            predt = self.booster.inplace_predict(X)
+            np.testing.assert_allclose(predt, predt_orig)
+
+        # unsupported types
+        for dtype in [
+            np.string_,
+            np.complex64,
+            np.complex128,
+        ]:
+            X = np.array(orig, dtype=dtype)
+            with pytest.raises(ValueError):
+                self.booster.inplace_predict(X)