scikit-learn-intelex 2024.5.0__py312-none-manylinux1_x86_64.whl → 2024.7.0__py312-none-manylinux1_x86_64.whl

{scikit_learn_intelex-2024.5.0.dist-info → scikit_learn_intelex-2024.7.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: scikit-learn-intelex
-Version: 2024.5.0
+Version: 2024.7.0
 Summary: Intel(R) Extension for Scikit-learn is a seamless way to speed up your Scikit-learn application.
 Home-page: https://github.com/intel/scikit-learn-intelex
 Author: Intel Corporation
@@ -31,7 +31,7 @@ Classifier: Topic :: Software Development
 Requires-Python: >=3.7
 Description-Content-Type: text/markdown
 License-File: LICENSE.txt
-Requires-Dist: daal4py (==2024.5.0)
+Requires-Dist: daal4py (==2024.7.0)
 Requires-Dist: scikit-learn (>=0.22)
 
 

scikit_learn_intelex-2024.7.0.dist-info/RECORD

@@ -0,0 +1,122 @@
+sklearnex/__init__.py,sha256=DVpGCMMQcUlrOGj_iy1jk3ZhCfa0_TFISYLqaUDRqC0,1780
+sklearnex/__main__.py,sha256=Le9BJq6aLEGSSoZwJLsOADxsV89ynBfA8BcoJHk9F24,1921
+sklearnex/_config.py,sha256=38wvKg2wG79ZctkvICl-Ay7hy8InXdR0t-P3n3h1NUs,3613
+sklearnex/_device_offload.py,sha256=h5lw46OS2shk5w3Nd-1Dk2eb4jV2eq5oTl3RtR7Xusc,3456
+sklearnex/_utils.py,sha256=EV4jC3plVdndsgrfPBsJZTzggrRdYWLwOpoIRWtTXt4,3812
+sklearnex/conftest.py,sha256=ODuhlscC0HNGXiA8olEfHTDULzjevqG9_sn0yMGRkHg,2376
+sklearnex/dispatcher.py,sha256=xxIcQ6YLSW37bvaDitlqE7qBeKfZx-c7cPAXGLNjbhk,18695
+sklearnex/basic_statistics/__init__.py,sha256=-lNh54F2FO0BuHDBLjhW7zp3yBcgFZ-DrXHj2H5hLXE,920
+sklearnex/basic_statistics/basic_statistics.py,sha256=1U_60TqIaNoWxA6XX1LQf18PP-1oOEjkyrskTBFBS7g,5482
+sklearnex/basic_statistics/incremental_basic_statistics.py,sha256=NNWF9zdxBz2jT7dzt5hL6HRvjQIGosvViQE3KJMF1L4,10048
+sklearnex/basic_statistics/tests/test_basic_statistics.py,sha256=dsJCHOPdjwZt5Kod1VdD8sdQrpoabPPUJNGN5S8cVvU,9533
+sklearnex/basic_statistics/tests/test_incremental_basic_statistics.py,sha256=0a5zE0iMzAQMiRyKTpcvQjGHBM86aNhnQZ0l_Ry1fNY,14914
+sklearnex/cluster/__init__.py,sha256=TsDzdbKzEubaP86iY8a7mSW0ZQzgjzGYuPkhc5lZmlQ,853
+sklearnex/cluster/dbscan.py,sha256=hpP-R2VTdeXjyoZkSw8AegdGtEa2kRt2KHIokZNHFXo,6824
+sklearnex/cluster/k_means.py,sha256=vo9NXztaphr3LWdsnwqPndycIbFsispSD2QldInyGRw,1262
+sklearnex/cluster/tests/test_dbscan.py,sha256=JYpwyuPkGQHdbE_IPbQv4qNj7clMm4UPdz_lrpRzKXE,1504
+sklearnex/cluster/tests/test_kmeans.py,sha256=TCPlrO0hrFkP3Aba_sHe2ZEAR0q16_YCLFMBBymQjdI,1643
+sklearnex/covariance/__init__.py,sha256=_c86wiKDGBH50ofN-_tn7lO4M2YVEWO5AHhJIfimUDk,859
+sklearnex/covariance/incremental_covariance.py,sha256=JnB6On_9MvDsNcDsSIb-4G9P38cx2A2qmomO3XjVi5I,13045
+sklearnex/covariance/tests/test_incremental_covariance.py,sha256=mhH66DSlqBweySB3HZ4a9GQhz_a8BngB0HQSQkJ5Oc8,8010
+sklearnex/decomposition/__init__.py,sha256=RJBYDWyvnn5DA-j5r4IqVdXLp6H4mdPySsSjnXhlv-U,805
+sklearnex/decomposition/pca.py,sha256=eoNighcTXtuVbH4KP1fBaHbAUUBLup2Bqbjd5fs_J1Q,16586
+sklearnex/decomposition/tests/test_pca.py,sha256=EoCgpSojE2S2e7hOUwW0Bh3vVGTUywawAhU7ThVAlW0,2319
+sklearnex/doc/third-party-programs.txt,sha256=qA1XbOqYkMRnp29p8VxXjfcH0kHE8NSO5s2heea83-8,21773
+sklearnex/ensemble/__init__.py,sha256=2xB2KfG7l7uJv0p3kfVhrs4KLM174SCsigZhYqwSYAA,1035
+sklearnex/ensemble/_forest.py,sha256=J3TN37gIIuWkPrVTY8xLKQNh4Xr3LSi7wAH7zCsl9X0,72353
+sklearnex/ensemble/tests/test_forest.py,sha256=uBpxWlhIOkaSzfd_nS6Ui0O7PcS-fVDzB5my2WQWOi0,4843
+sklearnex/glob/__main__.py,sha256=--FAjkhh5E6a1TFOGu6_BM3MHe78jP8oSpl0xKiRtTI,2531
+sklearnex/glob/dispatcher.py,sha256=o6XKGKM3M91F7FlXBOt1IhnpWQK4R1VY2WS-0uIghcw,3906
+sklearnex/linear_model/__init__.py,sha256=mzZ0EaBhDH66ETNt2vylznSoZbCYexgL2qE_jKppYYc,1144
+sklearnex/linear_model/coordinate_descent.py,sha256=h7ATAXbLTPisnRRQjt4nZiFzRjR53F2M_XSeZi_pFNw,1547
+sklearnex/linear_model/incremental_linear.py,sha256=iDad4Z4xMDImvG-Jgtt0Hz3Qp9MVhSi-IjUPcV0Y_ks,16101
+sklearnex/linear_model/linear.py,sha256=BDexE22Nl6egzHoMiruBp6L_biUfy0eWaZWbHMG9rJc,10740
+sklearnex/linear_model/logistic_path.py,sha256=Nq3JPXSzE4bjamnP3gGQtsMKks2v7s6bSiuYRnFqrRw,849
+sklearnex/linear_model/logistic_regression.py,sha256=e7tZhQX6URfoyuZ5OweiGGgTY19vfMrGGH02qiTXT0M,14881
+sklearnex/linear_model/ridge.py,sha256=ZrBcYGylPMRqFj3jUvJ7nQAioh8YbvCd8CttJlx1RgM,1183
+sklearnex/linear_model/tests/test_incremental_linear.py,sha256=e0ZADjB0myq1QcdwYxlVYl6tGFs4tVZIfBoV1xkdFuw,7337
+sklearnex/linear_model/tests/test_linear.py,sha256=7PN4sCZ0bdcknKPdxoV0WtFg__-GHm_0PeNVOAz6HGY,5553
+sklearnex/linear_model/tests/test_logreg.py,sha256=h_5X-IeKDE7TOSyCiP05WEa4ZbnZIQZ66Nwz9r8NzdM,4881
+sklearnex/manifold/__init__.py,sha256=3FU5fY1YvHAyNeSlUHsZfPnAbvlAUtZpum0Obmok2KE,809
+sklearnex/manifold/t_sne.py,sha256=EsyViHWIGt5vzHDUkGXBF3LGO6g10503ENLIHvdiv_Q,1005
+sklearnex/manifold/tests/test_tsne.py,sha256=SNqe1yg45mL4qIsxpX5hDpBsIyrv9r__6wfrerjp3yU,1063
+sklearnex/metrics/__init__.py,sha256=EHOCTU9RERa0Rrkr4dgmyO9ippbN5kjH_FwK3BqIROc,907
+sklearnex/metrics/pairwise.py,sha256=OVbb4F-NdHDIfxf9ThkZXD1wqIaFTOZ1J-_EEYIlv90,980
+sklearnex/metrics/ranking.py,sha256=ktFsbK6ILIQTwHM134KOf5MNTo0DoJYzfxV8GMHjRr8,957
+sklearnex/metrics/tests/test_metrics.py,sha256=Nwfz8UV4V4fKLLY7f9P9eg8uY09xLXaFx1GC49TOiA8,1579
+sklearnex/model_selection/__init__.py,sha256=64045Y-nzLDBepO6IRjt88LhL2DM3KdvpCF2vvj_RpA,842
+sklearnex/model_selection/split.py,sha256=IdUeCJghfifj5n2VCoxXfbHtIIRBaimyOEixB_UUKbM,974
+sklearnex/model_selection/tests/test_model_selection.py,sha256=3kZIq8kba8SW93DBWXupoRymNStR1_mGGYSErQRnwME,1338
+sklearnex/neighbors/__init__.py,sha256=_GMJ2L-6YechRYI2fOFwHjy0kebUncE81z4FmdXUlb8,1078
+sklearnex/neighbors/_lof.py,sha256=AtmBVdYBUpk7kqDNXbtvBNWkGNcmE0Lq_VtFgjelwhE,9120
+sklearnex/neighbors/common.py,sha256=G79Ifl656rSOHDoBWhBMoqBUer65BOeLFWo9ttdEpEU,12527
+sklearnex/neighbors/knn_classification.py,sha256=EJAColAfBUpUuBxBhWD3XmMDFqlPcgIotQa4RoJsdfU,7835
+sklearnex/neighbors/knn_regression.py,sha256=eKVqlEB-S2dPgeq8BOzaZwqPnKR2ZNeUBlvJkSSCUlg,6779
+sklearnex/neighbors/knn_unsupervised.py,sha256=GhvVlvcarq681OYHkmWqJ16z62HW_kAkDsrf6KErXmU,6067
+sklearnex/neighbors/tests/test_neighbors.py,sha256=yAlMnLt9GrdT6Ceph5B7iFuMJXpDURiHWTE99oO8EDw,3417
+sklearnex/preview/__init__.py,sha256=hZfIgTkkkUVaQ-SKaqI-S_SiXCkUzCUYxpSnbrhhEJU,813
+sklearnex/preview/cluster/__init__.py,sha256=FONzOuTGAb5NwdfFhLco99P_VccRk9NBkDHU3EKuAIs,794
+sklearnex/preview/cluster/_common.py,sha256=bgpzijxgyrAtdNCCHyCNsZ-N70KQYNwMuoCTNdUna6o,2718
+sklearnex/preview/cluster/k_means.py,sha256=RdF9mwgvZ1xgyeLhCQA2WaPvPsS1ndktXXcaNj6S460,10464
+sklearnex/preview/covariance/__init__.py,sha256=DPYTk8lwcVphtwU8J3CyUYH8Uz6Zr0Uz4S6nn-RY5iM,825
+sklearnex/preview/covariance/covariance.py,sha256=NjRTWLie7UbpfN16wGSxJSRHRhUnvTn03cPPc8PpPd8,5131
+sklearnex/preview/covariance/tests/test_covariance.py,sha256=FOgvhJxFQo9sNG4Uo-ebuw3ZXJ9tSxdk8qKm0ZJzTdc,2487
+sklearnex/preview/decomposition/__init__.py,sha256=9VcJPWKgSrWDFEXUY6ZCpAT2XGbOVA4a1j_XgfJBnTM,839
+sklearnex/preview/decomposition/incremental_pca.py,sha256=SsY1-AQt0mFTJGP5yzVxZvopNz2tSeXUO9p9c_3uVus,7820
+sklearnex/preview/decomposition/tests/test_incremental_pca.py,sha256=tsAlM18nzfIQxic7Ry986Ue0ovUdbopWFNckqQLK5xU,10776
+sklearnex/preview/linear_model/__init__.py,sha256=azZix9bU9zjwl12g0gB_K5RiVPJvkrinHHCwqVTLQLY,792
+sklearnex/preview/linear_model/ridge.py,sha256=SmUYSeBrBALJQUxsD3h4bWxuhU4oO7JhLGNnS3hxb1w,15343
+sklearnex/preview/linear_model/tests/test_ridge.py,sha256=Ru6D9fY2o6bXyxSeUo6O7sHF8jKHX30J545yBk0LRSo,4056
+sklearnex/spmd/__init__.py,sha256=ChQy2kEWlo4KGvs0RnbPoPEVhdgl8URV099B1rZtF5Y,889
+sklearnex/spmd/basic_statistics/__init__.py,sha256=NA5RGlwcp27UEeCgz0ngzYYd3MAIRpxlTy86uhM2-RE,821
+sklearnex/spmd/basic_statistics/basic_statistics.py,sha256=_dQ9mhVYxeuChATEpAmHpXDpgW3YvtK1qrG-kLr2MtI,886
+sklearnex/spmd/basic_statistics/tests/test_basic_statistics_spmd.py,sha256=I-VV-JTCR7Pu5Z36XtRxoLY6EUoeNp62DmM0vPu-NGY,3787
+sklearnex/spmd/cluster/__init__.py,sha256=qBBfrCHh6_82EROLbu54XKk7SmmRwS1XJyCj0zwkoUw,1029
+sklearnex/spmd/cluster/dbscan.py,sha256=23YNzPhx4MZijU0md-E3ZkHpTkhUh5cmtS3loHe-KhI,1824
+sklearnex/spmd/cluster/kmeans.py,sha256=Rnb9tr9LXVto5vCAumk7ZJfa9BYYDhdD1qUWL-QK5bY,868
+sklearnex/spmd/cluster/tests/test_dbscan_spmd.py,sha256=lGh7tWhoQaYMrwS3_BWcUkX2-GjsstWjLRw4JxMQgGA,3494
+sklearnex/spmd/cluster/tests/test_kmeans_spmd.py,sha256=ll4WL1TJDuLg-t3-Qvcl18J9uZLBJHEraf5jH3HQBKQ,5857
+sklearnex/spmd/covariance/__init__.py,sha256=5xeL1REMIxCv5M1ya99GGKaVjctUngboZ3uXgxcZ04o,823
+sklearnex/spmd/covariance/covariance.py,sha256=_oIlr1W1vqDHqIPnsCb04HcBCen5oBHQr5-_n9OSvIA,884
+sklearnex/spmd/covariance/tests/test_covariance_spmd.py,sha256=xH4jZ2F77KqVGYK5fyHTM28xNew7qN814lHAfrY1f50,3918
+sklearnex/spmd/decomposition/__init__.py,sha256=dBh0ZMIiaqdf3DKbt8FWNB2K9Iacs395m8OxaDFQg_M,784
+sklearnex/spmd/decomposition/pca.py,sha256=CUrsVD2jae-A9H8RB_emza_fe82CwnFa5PEy0fW_EZ8,871
+sklearnex/spmd/decomposition/tests/test_pca_spmd.py,sha256=9OktZfMTGz5fZfPLBSdOcDKtkcKl7I06CbLnqDajtW4,4709
+sklearnex/spmd/ensemble/__init__.py,sha256=B3yi7hWoVogMIiw0QRT_x5atsAFS-OO72YPLGeUQJ8M,873
+sklearnex/spmd/ensemble/forest.py,sha256=vIJJd10wIigFMtJMsiNDNLJ_PP2om-60zpQY_fd11-U,2909
+sklearnex/spmd/ensemble/tests/test_forest_spmd.py,sha256=YOE2sDXWvVHAPj4sswc8iPuWgnXzUSl_AKXDlN1w4zM,9245
+sklearnex/spmd/linear_model/__init__.py,sha256=WwqCr2DOyUnkSIHlP0yq0UI2yFDgSl5MHV7wu3QGJtA,893
+sklearnex/spmd/linear_model/linear_model.py,sha256=7QPCIQTWKBiZBTDZZbpZXi-REgxQCfRMt6rHPJAnc5E,883
+sklearnex/spmd/linear_model/logistic_regression.py,sha256=q_HkfWcg0RgFbk2t9FeV0ZY28HHAOtkGEnUj4tLuwt4,885
+sklearnex/spmd/linear_model/tests/test_linear_regression_spmd.py,sha256=l9bC5gDTNAS2QqZwUAgdbgG0GPvvQOT9XuyqajoAvic,5202
+sklearnex/spmd/linear_model/tests/test_logistic_regression_spmd.py,sha256=NweDwFyElY5HSwGRLH8IPNFqSnvynPCvQ97OFFe6Z68,6187
+sklearnex/spmd/neighbors/__init__.py,sha256=S16sH8N_18LN_8AgC_wGK5EDSNyuN-F-gI6GVlaiWVE,906
+sklearnex/spmd/neighbors/neighbors.py,sha256=SiKAS_RVt34MUcGytBS5pHnI_5vFNxJn8jqt1MOhDh8,940
+sklearnex/spmd/neighbors/tests/test_neighbors_spmd.py,sha256=aetHv4hdun2FaaHJTokCLc02szhkp0Yi552EdfboWeA,10488
+sklearnex/svm/__init__.py,sha256=f3e4ZFwZfx6MsXsn94VK1xVm6mWKT5XCiHczo6zNyAQ,1057
+sklearnex/svm/_common.py,sha256=2ml2e44mWvAufo6wCoJX3w6RbNON3OnyCaykZX0nkr8,12700
+sklearnex/svm/nusvc.py,sha256=-SfYz2bUMLaosYg_qeTBYf1Ra-bFuGmI62xIerf3XeE,11262
+sklearnex/svm/nusvr.py,sha256=9nbRaPCoDQ5SBGH_jLgcaX_PWGDh667w-gW5X2YbHBM,4764
+sklearnex/svm/svc.py,sha256=fZ9oVpKkgH5OYBXo4mPdZ-8cP07hSzPHoJsyItpXmcQ,12470
+sklearnex/svm/svr.py,sha256=NIWVUu7NjV4Fe4L-Y4vIqd3KePFdMJ1-jupeH4ZvNsg,4733
+sklearnex/svm/tests/test_svm.py,sha256=KnjWVfmHzU0sJqlhDdfLdhFJA_BV_tULPqNlOLXShXg,4194
+sklearnex/tests/_utils.py,sha256=rkZhH5VbMAIBBLBm2Mi5y5DQ_wMXQUq2LqR5X4wcEK0,10556
+sklearnex/tests/_utils_spmd.py,sha256=Kj6TdQXHyCUcI43H2qjcySyuHt4oy6OEqvX8BC8BiyM,6909
+sklearnex/tests/test_common.py,sha256=MYx0y7oomArxWu9qe2zNs7YL-ScPGFnzvw8PZ1or04A,1813
+sklearnex/tests/test_config.py,sha256=9ek00NtpWSU7HXNqXXM1kFX09FFS8DCHejVIFMii6kM,1540
+sklearnex/tests/test_memory_usage.py,sha256=FTtQgaa7p5ScDGscB93LkpU4B3DzfqnVMi84b2vSL30,10957
+sklearnex/tests/test_monkeypatch.py,sha256=Qeq0Z84UMKLEAVoQP7WEScouf6gNPbp21fjQ2erxMQo,9962
+sklearnex/tests/test_n_jobs_support.py,sha256=NziTP4GwZEDoBe1CDvhHZnp8JpwjYQmCNvXEDfS7Wo4,4313
+sklearnex/tests/test_parallel.py,sha256=0zzlh2VJZWcHw5W4QSUthtAygOb6K0Bg1ekNsrdGJQE,1770
+sklearnex/tests/test_patching.py,sha256=_-0IZ8n0tim0GRpZECzecOPebUUVmfKBA0yOUc7hbCw,14958
+sklearnex/tests/test_run_to_run_stability.py,sha256=Q4O7l1_0O4tZ_SM3kW1OxcOCuB73ugX_I4GZMi3Lrdg,10735
+sklearnex/utils/__init__.py,sha256=FFlM8zc2qnSdPYpLjSt4Ma_9FjC8qk8wXJa-5B8hCs0,898
+sklearnex/utils/_namespace.py,sha256=ohr8gOgEFgrccLecllMVYQPqbqqyye9uT-cWLtyxHFs,3167
+sklearnex/utils/parallel.py,sha256=VBcS-KUdyq7XpJUN6ygmNjyWtYLroghbvCxQ8nVU3YI,2085
+sklearnex/utils/validation.py,sha256=fjfhQiKnBQnD7LCBlacMyvsrhGnlMLRXk5Q69uoZIP4,827
+sklearnex/utils/tests/test_finite.py,sha256=AzJRY71X0VvDUicUI8Ey9Le6_yKp5O-3ZikhDVJNWms,2943
+scikit_learn_intelex-2024.7.0.dist-info/LICENSE.txt,sha256=7micbUpzQXphq9e_2oL7PpZcvoXzPuQHIDEXyKXC81s,10797
+scikit_learn_intelex-2024.7.0.dist-info/METADATA,sha256=3bnFVcELT6fXGuuN7F6toAcw45_HPE_sh7s-_mvdnPQ,12449
+scikit_learn_intelex-2024.7.0.dist-info/WHEEL,sha256=wKn5R5EEXodawgFUpV0Un8G_w3mugUfWciAZCmvm4qc,108
+scikit_learn_intelex-2024.7.0.dist-info/top_level.txt,sha256=kzKChSWGJEYFmdj5PwE63HNuP_PVOhWfD32ytH9rL9Q,10
+scikit_learn_intelex-2024.7.0.dist-info/RECORD,,

sklearnex/_config.py

@@ -14,24 +14,12 @@
 # limitations under the License.
 # ==============================================================================
 
-import threading
 from contextlib import contextmanager
 
 from sklearn import get_config as skl_get_config
 from sklearn import set_config as skl_set_config
 
-_default_global_config = {
-    "target_offload": "auto",
-    "allow_fallback_to_host": False,
-}
-
-_threadlocal = threading.local()
-
-
-def _get_sklearnex_threadlocal_config():
-    if not hasattr(_threadlocal, "global_config"):
-        _threadlocal.global_config = _default_global_config.copy()
-    return _threadlocal.global_config
+from onedal._config import _get_config as onedal_get_config
 
 
 def get_config():
@@ -46,7 +34,7 @@ def get_config():
     set_config : Set global configuration.
     """
     sklearn = skl_get_config()
-    sklearnex = _get_sklearnex_threadlocal_config().copy()
+    sklearnex = onedal_get_config()
     return {**sklearn, **sklearnex}
 
 
@@ -70,7 +58,7 @@ def set_config(target_offload=None, allow_fallback_to_host=None, **sklearn_confi
     """
     skl_set_config(**sklearn_configs)
 
-    local_config = _get_sklearnex_threadlocal_config()
+    local_config = onedal_get_config(copy=False)
 
     if target_offload is not None:
         local_config["target_offload"] = target_offload

sklearnex/_device_offload.py

@@ -14,141 +14,21 @@
 # limitations under the License.
 # ==============================================================================
 
-import logging
-import sys
-from collections.abc import Iterable
 from functools import wraps
 
-import numpy as np
+from onedal._device_offload import (
+    _copy_to_usm,
+    _get_global_queue,
+    _transfer_to_host,
+    dpnp_available,
+)
 
-try:
-    from dpctl import SyclQueue
-    from dpctl.memory import MemoryUSMDevice, as_usm_memory
-    from dpctl.tensor import usm_ndarray
-
-    dpctl_available = True
-except ImportError:
-    dpctl_available = False
-
-try:
+if dpnp_available:
     import dpnp
+    from onedal._device_offload import _convert_to_dpnp
 
-    dpnp_available = True
-except ImportError:
-    dpnp_available = False
 
 from ._config import get_config
-from ._utils import get_patch_message
-
-oneapi_is_available = "daal4py.oneapi" in sys.modules
-if oneapi_is_available:
-    from daal4py.oneapi import _get_device_name_sycl_ctxt, _get_sycl_ctxt_params
-
-
-class DummySyclQueue:
-    """This class is designed to act like dpctl.SyclQueue
-    to allow device dispatching in scenarios when dpctl is not available"""
-
-    class DummySyclDevice:
-        def __init__(self, filter_string):
-            self._filter_string = filter_string
-            self.is_cpu = "cpu" in filter_string
-            self.is_gpu = "gpu" in filter_string
-            # TODO: check for possibility of fp64 support
-            # on other devices in this dummy class
-            self.has_aspect_fp64 = self.is_cpu
-
-            if not (self.is_cpu):
-                logging.warning(
-                    "Device support is limited. "
-                    "Please install dpctl for full experience"
-                )
-
-        def get_filter_string(self):
-            return self._filter_string
-
-    def __init__(self, filter_string):
-        self.sycl_device = self.DummySyclDevice(filter_string)
-
-
-def _get_device_info_from_daal4py():
-    if oneapi_is_available:
-        return _get_device_name_sycl_ctxt(), _get_sycl_ctxt_params()
-    return None, dict()
-
-
-def _get_global_queue():
-    target = get_config()["target_offload"]
-    d4p_target, _ = _get_device_info_from_daal4py()
-    if d4p_target == "host":
-        d4p_target = "cpu"
-
-    QueueClass = DummySyclQueue if not dpctl_available else SyclQueue
-
-    if target != "auto":
-        if d4p_target is not None and d4p_target != target:
-            if not isinstance(target, str):
-                if d4p_target not in target.sycl_device.get_filter_string():
-                    raise RuntimeError(
-                        "Cannot use target offload option "
-                        "inside daal4py.oneapi.sycl_context"
-                    )
-            else:
-                raise RuntimeError(
-                    "Cannot use target offload option "
-                    "inside daal4py.oneapi.sycl_context"
-                )
-        if isinstance(target, QueueClass):
-            return target
-        return QueueClass(target)
-    if d4p_target is not None:
-        return QueueClass(d4p_target)
-    return None
-
-
-def _transfer_to_host(queue, *data):
-    has_usm_data, has_host_data = False, False
-
-    host_data = []
-    for item in data:
-        usm_iface = getattr(item, "__sycl_usm_array_interface__", None)
-        if usm_iface is not None:
-            if not dpctl_available:
-                raise RuntimeError(
-                    "dpctl need to be installed to work "
-                    "with __sycl_usm_array_interface__"
-                )
-            if queue is not None:
-                if queue.sycl_device != usm_iface["syclobj"].sycl_device:
-                    raise RuntimeError(
-                        "Input data shall be located " "on single target device"
-                    )
-            else:
-                queue = usm_iface["syclobj"]
-
-            buffer = as_usm_memory(item).copy_to_host()
-            order = "C"
-            if usm_iface["strides"] is not None:
-                if usm_iface["strides"][0] < usm_iface["strides"][1]:
-                    order = "F"
-            item = np.ndarray(
-                shape=usm_iface["shape"],
-                dtype=usm_iface["typestr"],
-                buffer=buffer,
-                order=order,
-            )
-            has_usm_data = True
-        else:
-            has_host_data = True
-
-        mismatch_host_item = usm_iface is None and item is not None and has_usm_data
-        mismatch_usm_item = usm_iface is not None and has_host_data
-
-        if mismatch_host_item or mismatch_usm_item:
-            raise RuntimeError("Input data shall be located on single target device")
-
-        host_data.append(item)
-    return queue, host_data
 
 
 def _get_backend(obj, queue, method_name, *data):
@@ -162,10 +42,7 @@ def _get_backend(obj, queue, method_name, *data):
         else:
             return "sklearn", None, patching_status
 
-    _, d4p_options = _get_device_info_from_daal4py()
-    allow_fallback_to_host = get_config()["allow_fallback_to_host"] or d4p_options.get(
-        "host_offload_on_fail", False
-    )
+    allow_fallback_to_host = get_config()["allow_fallback_to_host"]
 
     if gpu_device:
         patching_status = obj._onedal_gpu_supported(method_name, *data)
@@ -203,42 +80,6 @@ def dispatch(obj, method_name, branches, *args, **kwargs):
     )
 
 
-def _copy_to_usm(queue, array):
-    if not dpctl_available:
-        raise RuntimeError(
-            "dpctl need to be installed to work " "with __sycl_usm_array_interface__"
-        )
-
-    if hasattr(array, "__array__"):
-
-        try:
-            mem = MemoryUSMDevice(array.nbytes, queue=queue)
-            mem.copy_from_host(array.tobytes())
-            return usm_ndarray(array.shape, array.dtype, buffer=mem)
-        except ValueError as e:
-            # ValueError will raise if device does not support the dtype
-            # retry with float32 (needed for fp16 and fp64 support issues)
-            # try again as float32, if it is a float32 just raise the error.
-            if array.dtype == np.float32:
-                raise e
-            return _copy_to_usm(queue, array.astype(np.float32))
-    else:
-        if isinstance(array, Iterable):
-            array = [_copy_to_usm(queue, i) for i in array]
-        return array
-
-
-if dpnp_available:
-
-    def _convert_to_dpnp(array):
-        if isinstance(array, usm_ndarray):
-            return dpnp.array(array, copy=False)
-        elif isinstance(array, Iterable):
-            for i in range(len(array)):
-                array[i] = _convert_to_dpnp(array[i])
-        return array
-
-
 def wrap_output_data(func):
     @wraps(func)
     def wrapper(self, *args, **kwargs):

sklearnex/basic_statistics/basic_statistics.py

@@ -14,4 +14,130 @@
 # limitations under the License.
 # ==============================================================================
 
-from onedal.basic_statistics import BasicStatistics
+import numpy as np
+from sklearn.base import BaseEstimator
+from sklearn.utils import check_array
+from sklearn.utils.validation import _check_sample_weight
+
+from daal4py.sklearn._n_jobs_support import control_n_jobs
+from daal4py.sklearn._utils import sklearn_check_version
+from onedal.basic_statistics import BasicStatistics as onedal_BasicStatistics
+
+from .._device_offload import dispatch
+from .._utils import PatchingConditionsChain
+
+
+@control_n_jobs(decorated_methods=["fit"])
+class BasicStatistics(BaseEstimator):
+    """
+    Estimator for basic statistics.
+    Allows to compute basic statistics for provided data.
+    Parameters
+    ----------
+    result_options: string or list, default='all'
+        List of statistics to compute
+
+    Attributes (are existing only if corresponding result option exists)
+    ----------
+        min : ndarray of shape (n_features,)
+            Minimum of each feature over all samples.
+        max : ndarray of shape (n_features,)
+            Maximum of each feature over all samples.
+        sum : ndarray of shape (n_features,)
+            Sum of each feature over all samples.
+        mean : ndarray of shape (n_features,)
+            Mean of each feature over all samples.
+        variance : ndarray of shape (n_features,)
+            Variance of each feature over all samples.
+        variation : ndarray of shape (n_features,)
+            Variation of each feature over all samples.
+        sum_squares : ndarray of shape (n_features,)
+            Sum of squares for each feature over all samples.
+        standard_deviation : ndarray of shape (n_features,)
+            Standard deviation of each feature over all samples.
+        sum_squares_centered : ndarray of shape (n_features,)
+            Centered sum of squares for each feature over all samples.
+        second_order_raw_moment : ndarray of shape (n_features,)
+            Second order moment of each feature over all samples.
+    """
+
+    def __init__(self, result_options="all"):
+        self.options = result_options
+
+    _onedal_basic_statistics = staticmethod(onedal_BasicStatistics)
+
+    def _save_attributes(self):
+        assert hasattr(self, "_onedal_estimator")
+
+        if self.options == "all":
+            result_options = onedal_BasicStatistics.get_all_result_options()
+        else:
+            result_options = self.options
+
+        if isinstance(result_options, str):
+            setattr(self, result_options, getattr(self._onedal_estimator, result_options))
+        elif isinstance(result_options, list):
+            for option in result_options:
+                setattr(self, option, getattr(self._onedal_estimator, option))
+
+    def _onedal_supported(self, method_name, *data):
+        patching_status = PatchingConditionsChain(
+            f"sklearnex.basic_statistics.{self.__class__.__name__}.{method_name}"
+        )
+        return patching_status
+
+    _onedal_cpu_supported = _onedal_supported
+    _onedal_gpu_supported = _onedal_supported
+
+    def _onedal_fit(self, X, sample_weight=None, queue=None):
+        if sklearn_check_version("1.0"):
+            X = self._validate_data(X, dtype=[np.float64, np.float32], ensure_2d=False)
+        else:
+            X = check_array(X, dtype=[np.float64, np.float32])
+
+        if sample_weight is not None:
+            sample_weight = _check_sample_weight(sample_weight, X)
+
+        onedal_params = {
+            "result_options": self.options,
+        }
+
+        if not hasattr(self, "_onedal_estimator"):
+            self._onedal_estimator = self._onedal_basic_statistics(**onedal_params)
+        self._onedal_estimator.fit(X, sample_weight, queue)
+        self._save_attributes()
+
+    def compute(self, data, weights=None, queue=None):
+        return self._onedal_estimator.compute(data, weights, queue)
+
+    def fit(self, X, y=None, *, sample_weight=None):
+        """Compute statistics with X, using minibatches of size batch_size.
+
+        Parameters
+        ----------
+        X : array-like of shape (n_samples, n_features)
+            Data for compute, where `n_samples` is the number of samples and
+            `n_features` is the number of features.
+
+        y : Ignored
+            Not used, present for API consistency by convention.
+
+        sample_weight : array-like of shape (n_samples,), default=None
+            Weights for compute weighted statistics, where `n_samples` is the number of samples.
+
+        Returns
+        -------
+        self : object
+            Returns the instance itself.
+        """
+        dispatch(
+            self,
+            "fit",
+            {
+                "onedal": self.__class__._onedal_fit,
+                "sklearn": None,
+            },
+            X,
+            sample_weight,
+        )
+        return self