Skip to content

Commit 234d7da

Browse files
remove details that are not applicable anymore (#3125)
1 parent 0e36592 commit 234d7da

1 file changed

Lines changed: 1 addition & 12 deletions

File tree

doc/sources/array_api.rst

Lines changed: 1 addition & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -43,8 +43,7 @@ requested operation (e.g. call to ``.fit()`` / ``.predict()``
4343
on the estimator class being used) is :ref:`supported on device/GPU <sklearn_algorithms_gpu>`, computations
4444
will be performed on the device where the data lives, without involving any data transfers.
4545

46-
Note that all of the inputs (e.g. ``X`` and ``y`` passed to ``.fit()`` methods) must be allocated on the same device for this to
47-
work. If the requested operation is not supported on the device where the data lives, then it will either fall
46+
If the requested operation is not supported on the device where the data lives, then it will either fall
4847
back to |sklearn|, or to an accelerated CPU version from the |sklearnex| when supported - these are controllable
4948
through options ``allow_sklearn_after_onedal`` (default is ``True``) and ``allow_fallback_to_host`` (default is
5049
``False``), respectively, which are accepted by ``config_context`` and ``set_config`` after
@@ -67,11 +66,6 @@ be transferred to host if it isn't already, and the computations will happen on
6766
Enable :ref:`verbose` to see information about whether data transfers happen during an operation or not,
6867
whether an accelerated version from the extension is used, and where (CPU/device) the operation is executed.
6968

70-
When passing array API inputs to methods such as ``.predict()`` of estimators with array API support, the output
71-
will always be of the same class as the inputs, but be aware that array attributes of fitted models (e.g. ``coef_``
72-
in a linear model) will not necessarily be of the same class as array API inputs passed to ``.fit()``, even though
73-
in many cases they are.
74-
7569
.. warning::
7670
If array API inputs are passed to an estimator's ``.fit()``, subsequent data passed to methods such as
7771
``.predict()`` or ``.score()`` of the fitted model **must reside on the same device** - meaning: a model that
@@ -128,11 +122,6 @@ The following patched classes have support for array API inputs:
128122
next subsection.
129123

130124
.. note::
131-
While full array API support is currently not implemented for all classes, |dpnp_array| inputs are supported
132-
by all the classes that have :ref:`GPU support <oneapi_gpu>`. Note however that if array API support is not
133-
enabled in |sklearn|, when passing these classes as inputs, data will be transferred to host and then back to
134-
device instead of being used directly.
135-
136125
Result attributes of |sklearnex| classes which contain |sklearn| or |sklearnex| classes may not themselves be
137126
array API compliant. For example, ensemble algorithms contain decision tree estimators result objects which
138127
do not comply with the array API standard.

0 commit comments

Comments
 (0)