Running of sklearn tests on device=cpu (#1203)

* Add CI step on lnx

* Add actual dpctl installation

* Pass exit code in run_sklearn_tests scripts

* Modify dpcpp packages installation

* Add missing 'has_aspect_fp64' attribute in DummySyclDevice

Author: Alexsandruss

.ci/pipeline/ci.yml

@@ -100,8 +100,8 @@ jobs:
     displayName: 'System info'
   - script: |
       conda update -y -q conda
-      if [ $(echo $(PYTHON_VERSION) | grep '3.11') ]; then export DPCTL_PACKAGE=; else export DPCTL_PACKAGE=; fi
-      conda create -q -y -n CB -c conda-forge -c intel python=$(PYTHON_VERSION) dal-devel mpich pyyaml $DPCTL_PACKAGE
+      if [ $(echo $(PYTHON_VERSION) | grep '3.7') ] || [ $(echo $(PYTHON_VERSION) | grep '3.11') ]; then export DPCPP_PACKAGE="dpcpp-cpp-rt>=2023.0.0"; else export DPCPP_PACKAGE="dpctl>=0.14"; fi
+      conda create -q -y -n CB -c conda-forge -c intel python=$(PYTHON_VERSION) dal-devel mpich pyyaml $DPCPP_PACKAGE
     displayName: 'Conda create'
   - script: |
       . /usr/share/miniconda/etc/profile.d/conda.sh
@@ -143,6 +143,11 @@ jobs:
       conda activate CB
       bash .ci/scripts/run_sklearn_tests.sh
     displayName: 'Sklearn testing'
+  - script: |
+      . /usr/share/miniconda/etc/profile.d/conda.sh
+      conda activate CB
+      bash .ci/scripts/run_sklearn_tests.sh cpu
+    displayName: 'Sklearn testing [device=cpu]'
   - script: |
       . /usr/share/miniconda/etc/profile.d/conda.sh
       conda activate CB

.ci/scripts/run_sklearn_tests.py

@@ -0,0 +1,48 @@
+#===============================================================================
+# Copyright 2023 Intel Corporation
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#===============================================================================
+
+from sklearnex import patch_sklearn
+patch_sklearn()
+
+import os
+import sys
+import argparse
+import pytest
+import sklearn
+
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        '-d', '--device',
+        type=str,
+        default='none',
+        help='device name',
+        choices=['none', 'cpu', 'gpu']
+    )
+    args = parser.parse_args()
+
+    os.chdir(os.path.dirname(sklearn.__file__))
+
+    pytest_args = '--verbose --pyargs --durations=100 --durations-min=0.01 ' \
+        f'{os.environ["DESELECTED_TESTS"]} {os.environ["SELECTED_TESTS"]}'.split(' ')
+
+    if args.device != 'none':
+        with sklearn.config_context(target_offload=args.device):
+            return_code = pytest.main(pytest_args)
+    else:
+        return_code = pytest.main(pytest_args)
+    sys.exit(int(return_code))

.ci/scripts/run_sklearn_tests.sh

@@ -15,11 +15,14 @@
 # limitations under the License.
 #===============================================================================
 
+# Args:
+# 1 - device name (optional)
+
 ci_dir=$( dirname $( dirname "${BASH_SOURCE[0]}" ) )
 cd $ci_dir
 
 export SELECTED_TESTS=$(python scripts/select_sklearn_tests.py)
 export DESELECTED_TESTS=$(python ../.circleci/deselect_tests.py ../deselected_tests.yaml)
-cd $(python -c "import sklearn, os; print(os.path.dirname(sklearn.__file__))")
-export SKLEARNEX_VERBOSE=DEBUG
-python -m sklearnex -m pytest --verbose --pyargs --durations=100 --durations-min=0.01 $DESELECTED_TESTS $SELECTED_TESTS
+
+python scripts/run_sklearn_tests.py -d ${1:-none}
+exit $?

sklearnex/_device_offload.py

@@ -43,6 +43,9 @@ 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):
                 import logging