Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

DRAFT: timeouts deselection on GPU deselection #2059

Closed
7 changes: 5 additions & 2 deletions .circleci/run_xpu_tests.py
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Just for debug; will be reverted.

Original file line number Diff line number Diff line change
Expand Up @@ -71,15 +71,18 @@

pytest_params = ["-ra", "--disable-warnings"]

if not args.quiet:
pytest_params.append("-q")
# Ignoring quiet flag now.
# if not args.quiet:
# pytest_params.append("-q")

if not args.no_intel_optimized:
from sklearnex import patch_sklearn

patch_sklearn()

if args.device == "gpu":
# Adding explicitly verbose for GPU testing
pytest_params.append("-vv")
from sklearnex._config import config_context

with config_context(target_offload=args.device, allow_fallback_to_host=False):
Expand Down
176 changes: 176 additions & 0 deletions deselected_tests.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -463,8 +463,184 @@ gpu:
- ensemble/tests/test_weight_boosting.py
# Fails
- cluster/tests/test_dbscan.py::test_weighted_dbscan
- cluster/tests/test_k_means.py::test_kmeans_results[float32-lloyd-sparse]
- cluster/tests/test_k_means.py::test_kmeans_results[float32-elkan-sparse]
- cluster/tests/test_k_means.py::test_kmeans_results[float64-lloyd-sparse]
- cluster/tests/test_k_means.py::test_kmeans_results[float64-elkan-sparse]
- cluster/tests/test_k_means.py::test_relocate_empty_clusters[sparse]
- cluster/tests/test_k_means.py::test_all_init[KMeans-random-sparse]
- cluster/tests/test_k_means.py::test_all_init[KMeans-k-means++-sparse]
- cluster/tests/test_k_means.py::test_all_init[KMeans-ndarray-sparse]
- cluster/tests/test_k_means.py::test_all_init[KMeans-callable-sparse]
- cluster/tests/test_k_means.py::test_all_init[MiniBatchKMeans-random-sparse]
- cluster/tests/test_k_means.py::test_all_init[MiniBatchKMeans-k-means++-sparse]
- cluster/tests/test_k_means.py::test_all_init[MiniBatchKMeans-ndarray-sparse]
- cluster/tests/test_k_means.py::test_all_init[MiniBatchKMeans-callable-sparse]
- cluster/tests/test_k_means.py::test_minibatch_reassign[42-sparse]
- cluster/tests/test_k_means.py::test_kmeans_predict[float32-42-2-KMeans-lloyd-sparse]
- cluster/tests/test_k_means.py::test_kmeans_predict[float32-42-2-KMeans-elkan-sparse]
- cluster/tests/test_k_means.py::test_kmeans_predict[float32-42-2-MiniBatchKMeans-None-sparse]
- cluster/tests/test_k_means.py::test_kmeans_predict[float32-42-100-KMeans-lloyd-sparse]
- cluster/tests/test_k_means.py::test_kmeans_predict[float32-42-100-KMeans-elkan-sparse]
- cluster/tests/test_k_means.py::test_kmeans_predict[float32-42-100-MiniBatchKMeans-None-sparse]
- cluster/tests/test_k_means.py::test_kmeans_predict[float64-42-2-KMeans-lloyd-sparse]
- cluster/tests/test_k_means.py::test_kmeans_predict[float64-42-2-KMeans-elkan-sparse]
- cluster/tests/test_k_means.py::test_kmeans_predict[float64-42-2-MiniBatchKMeans-None-sparse]
- cluster/tests/test_k_means.py::test_kmeans_predict[float64-42-100-KMeans-lloyd-sparse]
- cluster/tests/test_k_means.py::test_kmeans_predict[float64-42-100-KMeans-elkan-sparse]
- cluster/tests/test_k_means.py::test_kmeans_predict[float64-42-100-MiniBatchKMeans-None-sparse]
- cluster/tests/test_k_means.py::test_dense_sparse[42-KMeans]
- cluster/tests/test_k_means.py::test_dense_sparse[42-MiniBatchKMeans]
- cluster/tests/test_k_means.py::test_predict_dense_sparse[KMeans-random]
- cluster/tests/test_k_means.py::test_predict_dense_sparse[KMeans-k-means++]
- cluster/tests/test_k_means.py::test_predict_dense_sparse[KMeans-ndarray]
- cluster/tests/test_k_means.py::test_predict_dense_sparse[MiniBatchKMeans-random]
- cluster/tests/test_k_means.py::test_predict_dense_sparse[MiniBatchKMeans-k-means++]
- cluster/tests/test_k_means.py::test_predict_dense_sparse[MiniBatchKMeans-ndarray]
- cluster/tests/test_k_means.py::test_integer_input[42-KMeans-k-means++-int32-sparse]
- cluster/tests/test_k_means.py::test_integer_input[42-KMeans-k-means++-int64-sparse]
- cluster/tests/test_k_means.py::test_integer_input[42-KMeans-ndarray-int32-sparse]
- cluster/tests/test_k_means.py::test_integer_input[42-KMeans-ndarray-int64-sparse]
- cluster/tests/test_k_means.py::test_integer_input[42-MiniBatchKMeans-k-means++-int32-sparse]
- cluster/tests/test_k_means.py::test_integer_input[42-MiniBatchKMeans-k-means++-int64-sparse]
- cluster/tests/test_k_means.py::test_integer_input[42-MiniBatchKMeans-ndarray-int32-sparse]
- cluster/tests/test_k_means.py::test_integer_input[42-MiniBatchKMeans-ndarray-int64-sparse]
- cluster/tests/test_k_means.py::test_float_precision[42-KMeans-sparse]
- cluster/tests/test_k_means.py::test_float_precision[42-MiniBatchKMeans-sparse]
- cluster/tests/test_k_means.py::test_kmeans_init_fitted_centers[sparse]
- cluster/tests/test_k_means.py::test_unit_weights_vs_no_weights[42-KMeans-sparse]
- cluster/tests/test_k_means.py::test_unit_weights_vs_no_weights[42-MiniBatchKMeans-sparse]
- cluster/tests/test_k_means.py::test_scaled_weights[42-KMeans-sparse]
- cluster/tests/test_k_means.py::test_scaled_weights[42-MiniBatchKMeans-sparse]
- cluster/tests/test_k_means.py::test_kmeans_empty_cluster_relocated[sparse]
- cluster/tests/test_k_means.py::test_k_means_1_iteration[42-lloyd-sparse]
- cluster/tests/test_k_means.py::test_k_means_1_iteration[42-elkan-sparse]
- cluster/tests/test_k_means.py::test_kmeans_elkan_results[42-0-sparse-normal]
- cluster/tests/test_k_means.py::test_kmeans_elkan_results[42-0-sparse-blobs]
- cluster/tests/test_k_means.py::test_kmeans_elkan_results[42-1e-100-sparse-normal]
- cluster/tests/test_k_means.py::test_kmeans_elkan_results[42-1e-100-sparse-blobs]
- cluster/tests/test_k_means.py::test_kmeans_elkan_results[42-0.01-sparse-normal]
- cluster/tests/test_k_means.py::test_kmeans_elkan_results[42-0.01-sparse-blobs]
- cluster/tests/test_k_means.py::test_kmeans_elkan_results[42-1e-08-sparse-normal]
- cluster/tests/test_k_means.py::test_kmeans_elkan_results[42-1e-08-sparse-blobs]
- cluster/tests/test_k_means.py::test_kmeans_elkan_results[42-0.01-sparse_matrix-normal]
- cluster/tests/test_k_means.py::test_kmeans_elkan_results[42-0.01-sparse_matrix-blobs]
- cluster/tests/test_k_means.py::test_kmeans_elkan_results[42-0.01-sparse_array-normal]
- cluster/tests/test_k_means.py::test_kmeans_elkan_results[42-0.01-sparse_array-blobs]
- cluster/tests/test_k_means.py::test_kmeans_elkan_results[42-1e-08-sparse_matrix-normal]
- cluster/tests/test_k_means.py::test_kmeans_elkan_results[42-1e-08-sparse_matrix-blobs]
- cluster/tests/test_k_means.py::test_kmeans_elkan_results[42-1e-08-sparse_array-normal]
- cluster/tests/test_k_means.py::test_kmeans_elkan_results[42-1e-08-sparse_array-blobs]
- cluster/tests/test_k_means.py::test_kmeans_elkan_results[42-1e-100-sparse_matrix-normal]
- cluster/tests/test_k_means.py::test_kmeans_elkan_results[42-1e-100-sparse_matrix-blobs]
- cluster/tests/test_k_means.py::test_kmeans_elkan_results[42-1e-100-sparse_array-normal]
- cluster/tests/test_k_means.py::test_kmeans_elkan_results[42-1e-100-sparse_array-blobs]
- cluster/tests/test_k_means.py::test_kmeans_elkan_results[42-0-sparse_matrix-normal]
- cluster/tests/test_k_means.py::test_kmeans_elkan_results[42-0-sparse_matrix-blobs]
- cluster/tests/test_k_means.py::test_kmeans_elkan_results[42-0-sparse_array-normal]
- cluster/tests/test_k_means.py::test_kmeans_elkan_results[42-0-sparse_array-blobs]
- cluster/tests/test_k_means.py::test_minibatch_update_consistency[42-X_csr0]
- cluster/tests/test_k_means.py::test_minibatch_update_consistency[42-X_csr1]
- cluster/tests/test_k_means.py::test_all_init[KMeans-random-sparse_matrix]
- cluster/tests/test_k_means.py::test_all_init[KMeans-random-sparse_array]
- cluster/tests/test_k_means.py::test_all_init[KMeans-k-means++-sparse_matrix]
- cluster/tests/test_k_means.py::test_all_init[KMeans-k-means++-sparse_array]
- cluster/tests/test_k_means.py::test_all_init[KMeans-ndarray-sparse_matrix]
- cluster/tests/test_k_means.py::test_all_init[KMeans-ndarray-sparse_array]
- cluster/tests/test_k_means.py::test_all_init[KMeans-callable-sparse_matrix]
- cluster/tests/test_k_means.py::test_all_init[KMeans-callable-sparse_array]
- cluster/tests/test_k_means.py::test_all_init[MiniBatchKMeans-random-sparse_matrix]
- cluster/tests/test_k_means.py::test_all_init[MiniBatchKMeans-random-sparse_array]
- cluster/tests/test_k_means.py::test_all_init[MiniBatchKMeans-k-means++-sparse_matrix]
- cluster/tests/test_k_means.py::test_all_init[MiniBatchKMeans-k-means++-sparse_array]
- cluster/tests/test_k_means.py::test_all_init[MiniBatchKMeans-ndarray-sparse_matrix]
- cluster/tests/test_k_means.py::test_all_init[MiniBatchKMeans-ndarray-sparse_array]
- cluster/tests/test_k_means.py::test_all_init[MiniBatchKMeans-callable-sparse_matrix]
- cluster/tests/test_k_means.py::test_all_init[MiniBatchKMeans-callable-sparse_array]
- cluster/tests/test_k_means.py::test_minibatch_reassign[42-sparse_matrix]
- cluster/tests/test_k_means.py::test_minibatch_reassign[42-sparse_array]
- cluster/tests/test_k_means.py::test_kmeans_predict[float32-42-2-KMeans-lloyd-sparse_matrix]
- cluster/tests/test_k_means.py::test_kmeans_predict[float32-42-2-KMeans-lloyd-sparse_array]
- cluster/tests/test_k_means.py::test_kmeans_predict[float32-42-2-KMeans-elkan-sparse_matrix]
- cluster/tests/test_k_means.py::test_kmeans_predict[float32-42-2-KMeans-elkan-sparse_array]
- cluster/tests/test_k_means.py::test_kmeans_predict[float32-42-2-MiniBatchKMeans-None-sparse_matrix]
- cluster/tests/test_k_means.py::test_kmeans_predict[float32-42-2-MiniBatchKMeans-None-sparse_array]
- cluster/tests/test_k_means.py::test_kmeans_predict[float32-42-100-KMeans-lloyd-sparse_matrix]
- cluster/tests/test_k_means.py::test_kmeans_predict[float32-42-100-KMeans-lloyd-sparse_array]
- cluster/tests/test_k_means.py::test_kmeans_predict[float32-42-100-KMeans-elkan-sparse_matrix]
- cluster/tests/test_k_means.py::test_kmeans_predict[float32-42-100-KMeans-elkan-sparse_array]
- cluster/tests/test_k_means.py::test_kmeans_predict[float32-42-100-MiniBatchKMeans-None-sparse_matrix]
- cluster/tests/test_k_means.py::test_kmeans_predict[float32-42-100-MiniBatchKMeans-None-sparse_array]
- cluster/tests/test_k_means.py::test_kmeans_predict[float64-42-2-KMeans-lloyd-sparse_matrix]
- cluster/tests/test_k_means.py::test_kmeans_predict[float64-42-2-KMeans-lloyd-sparse_array]
- cluster/tests/test_k_means.py::test_kmeans_predict[float64-42-2-KMeans-elkan-sparse_matrix]
- cluster/tests/test_k_means.py::test_kmeans_predict[float64-42-2-KMeans-elkan-sparse_array]
- cluster/tests/test_k_means.py::test_kmeans_predict[float64-42-2-MiniBatchKMeans-None-sparse_matrix]
- cluster/tests/test_k_means.py::test_kmeans_predict[float64-42-2-MiniBatchKMeans-None-sparse_array]
- cluster/tests/test_k_means.py::test_kmeans_predict[float64-42-100-KMeans-lloyd-sparse_matrix]
- cluster/tests/test_k_means.py::test_kmeans_predict[float64-42-100-KMeans-lloyd-sparse_array]
- cluster/tests/test_k_means.py::test_kmeans_predict[float64-42-100-KMeans-elkan-sparse_matrix]
- cluster/tests/test_k_means.py::test_kmeans_predict[float64-42-100-KMeans-elkan-sparse_array]
- cluster/tests/test_k_means.py::test_kmeans_predict[float64-42-100-MiniBatchKMeans-None-sparse_matrix]
- cluster/tests/test_k_means.py::test_kmeans_predict[float64-42-100-MiniBatchKMeans-None-sparse_array]
- cluster/tests/test_k_means.py::test_dense_sparse[42-KMeans-X_csr0]
- cluster/tests/test_k_means.py::test_dense_sparse[42-KMeans-X_csr1]
- cluster/tests/test_k_means.py::test_dense_sparse[42-MiniBatchKMeans-X_csr0]
- cluster/tests/test_k_means.py::test_dense_sparse[42-MiniBatchKMeans-X_csr1]
- cluster/tests/test_k_means.py::test_predict_dense_sparse[KMeans-random-X_csr0]
- cluster/tests/test_k_means.py::test_predict_dense_sparse[KMeans-random-X_csr1]
- cluster/tests/test_k_means.py::test_predict_dense_sparse[KMeans-k-means++-X_csr0]
- cluster/tests/test_k_means.py::test_predict_dense_sparse[KMeans-k-means++-X_csr1]
- cluster/tests/test_k_means.py::test_predict_dense_sparse[KMeans-ndarray-X_csr0]
- cluster/tests/test_k_means.py::test_predict_dense_sparse[KMeans-ndarray-X_csr1]
- cluster/tests/test_k_means.py::test_predict_dense_sparse[MiniBatchKMeans-random-X_csr0]
- cluster/tests/test_k_means.py::test_predict_dense_sparse[MiniBatchKMeans-random-X_csr1]
- cluster/tests/test_k_means.py::test_predict_dense_sparse[MiniBatchKMeans-k-means++-X_csr0]
- cluster/tests/test_k_means.py::test_predict_dense_sparse[MiniBatchKMeans-k-means++-X_csr1]
- cluster/tests/test_k_means.py::test_predict_dense_sparse[MiniBatchKMeans-ndarray-X_csr0]
- cluster/tests/test_k_means.py::test_predict_dense_sparse[MiniBatchKMeans-ndarray-X_csr1]
- cluster/tests/test_k_means.py::test_integer_input[42-KMeans-k-means++-int32-sparse_matrix]
- cluster/tests/test_k_means.py::test_integer_input[42-KMeans-k-means++-int32-sparse_array]
- cluster/tests/test_k_means.py::test_integer_input[42-KMeans-k-means++-int64-sparse_matrix]
- cluster/tests/test_k_means.py::test_integer_input[42-KMeans-k-means++-int64-sparse_array]
- cluster/tests/test_k_means.py::test_integer_input[42-KMeans-ndarray-int32-sparse_matrix]
- cluster/tests/test_k_means.py::test_integer_input[42-KMeans-ndarray-int32-sparse_array]
- cluster/tests/test_k_means.py::test_integer_input[42-KMeans-ndarray-int64-sparse_matrix]
- cluster/tests/test_k_means.py::test_integer_input[42-KMeans-ndarray-int64-sparse_array]
- cluster/tests/test_k_means.py::test_integer_input[42-MiniBatchKMeans-k-means++-int32-sparse_matrix]
- cluster/tests/test_k_means.py::test_integer_input[42-MiniBatchKMeans-k-means++-int32-sparse_array]
- cluster/tests/test_k_means.py::test_integer_input[42-MiniBatchKMeans-k-means++-int64-sparse_matrix]
- cluster/tests/test_k_means.py::test_integer_input[42-MiniBatchKMeans-k-means++-int64-sparse_array]
- cluster/tests/test_k_means.py::test_integer_input[42-MiniBatchKMeans-ndarray-int32-sparse_matrix]
- cluster/tests/test_k_means.py::test_integer_input[42-MiniBatchKMeans-ndarray-int32-sparse_array]
- cluster/tests/test_k_means.py::test_integer_input[42-MiniBatchKMeans-ndarray-int64-sparse_matrix]
- cluster/tests/test_k_means.py::test_integer_input[42-MiniBatchKMeans-ndarray-int64-sparse_array]
- cluster/tests/test_k_means.py::test_float_precision[42-KMeans-sparse_matrix]
- cluster/tests/test_k_means.py::test_float_precision[42-KMeans-sparse_array]
- cluster/tests/test_k_means.py::test_float_precision[42-MiniBatchKMeans-sparse_matrix]
- cluster/tests/test_k_means.py::test_float_precision[42-MiniBatchKMeans-sparse_array]
- cluster/tests/test_k_means.py::test_kmeans_init_fitted_centers[sparse_matrix]
- cluster/tests/test_k_means.py::test_kmeans_init_fitted_centers[sparse_array]
- cluster/tests/test_k_means.py::test_unit_weights_vs_no_weights[42-KMeans-sparse_matrix]
- cluster/tests/test_k_means.py::test_unit_weights_vs_no_weights[42-KMeans-sparse_array]
- cluster/tests/test_k_means.py::test_unit_weights_vs_no_weights[42-MiniBatchKMeans-sparse_matrix]
- cluster/tests/test_k_means.py::test_unit_weights_vs_no_weights[42-MiniBatchKMeans-sparse_array]
- cluster/tests/test_k_means.py::test_scaled_weights[42-KMeans-sparse_matrix]
- cluster/tests/test_k_means.py::test_scaled_weights[42-KMeans-sparse_array]
- cluster/tests/test_k_means.py::test_scaled_weights[42-MiniBatchKMeans-sparse_matrix]
- cluster/tests/test_k_means.py::test_scaled_weights[42-MiniBatchKMeans-sparse_array]
- cluster/tests/test_k_means.py::test_kmeans_empty_cluster_relocated[sparse_matrix]
- cluster/tests/test_k_means.py::test_kmeans_empty_cluster_relocated[sparse_array]
- cluster/tests/test_k_means.py::test_k_means_1_iteration[42-lloyd-sparse_matrix]
- cluster/tests/test_k_means.py::test_k_means_1_iteration[42-lloyd-sparse_array]
- cluster/tests/test_k_means.py::test_k_means_1_iteration[42-elkan-sparse_matrix]
- cluster/tests/test_k_means.py::test_k_means_1_iteration[42-elkan-sparse_array]
- cluster/tests/test_k_means.py::test_relocating_with_duplicates[lloyd-sparse_matrix]
- cluster/tests/test_k_means.py::test_relocating_with_duplicates[lloyd-sparse_array]
- cluster/tests/test_k_means.py::test_relocating_with_duplicates[elkan-sparse_matrix]
- cluster/tests/test_k_means.py::test_relocating_with_duplicates[elkan-sparse_array]
- cluster/tests/test_k_means.py::test_predict_does_not_change_cluster_centers[csr_matrix]
- cluster/tests/test_k_means.py::test_predict_does_not_change_cluster_centers[csr_array]
- model_selection/tests/test_search.py::test_unsupervised_grid_search

- ensemble/tests/test_bagging.py::test_gridsearch
Expand Down