Skip to content

Add array_api_dispatch to spmd validation scope (backport #2940)#3098

Merged
maria-Petrova merged 1 commit intorls/2026.0.0-rlsfrom
mergify/bp/rls/2026.0.0-rls/pr-2940
Apr 8, 2026
Merged

Add array_api_dispatch to spmd validation scope (backport #2940)#3098
maria-Petrova merged 1 commit intorls/2026.0.0-rlsfrom
mergify/bp/rls/2026.0.0-rls/pr-2940

Conversation

@mergify
Copy link
Copy Markdown

@mergify mergify Bot commented Apr 3, 2026

Description

Additionally, adds array_api_dispatch as a parameter for SPMD tests, similarly to how use_raw_inputs was configured previously.


Checklist:

Completeness and readability

  • I have commented my code, particularly in hard-to-understand areas.
  • I have updated the documentation to reflect the changes or created a separate PR with updates and provided its number in the description, if necessary.
  • Git commit message contains an appropriate signed-off-by string (see CONTRIBUTING.md for details).
  • I have resolved any merge conflicts that might occur with the base branch.

Testing

  • I have run it locally and tested the changes extensively.
  • All CI jobs are green or I have provided justification why they aren't.
  • I have extended testing suite if new functionality was introduced in this PR.

Performance

  • I have measured performance for affected algorithms using scikit-learn_bench and provided at least a summary table with measured data, if performance change is expected.
  • I have provided justification why performance and/or quality metrics have changed or why changes are not expected.
  • I have extended the benchmarking suite and provided a corresponding scikit-learn_bench PR if new measurable functionality was introduced in this PR.

This is an automatic backport of pull request #2940 done by [Mergify](https://mergify.com).

* Add array_api_dispatch to spmd validation scope

* formatting

* Combine onedal _device_offload.py change

* re-add transform output logic

* black

* minor restoration

* restore host transfer if no queue

* add new array api algos and import config_context

* debugging

* kmeans hotfix

* remove debug

(cherry picked from commit 29ec7ad)
@codecov
Copy link
Copy Markdown

codecov Bot commented Apr 3, 2026

Codecov Report

❌ Patch coverage is 61.53846% with 5 lines in your changes missing coverage. Please review.

Files with missing lines Patch % Lines
onedal/_device_offload.py 61.53% 3 Missing and 2 partials ⚠️
Flag Coverage Δ
azure 79.63% <61.53%> (-0.04%) ⬇️
github ?

Flags with carried forward coverage won't be shown. Click here to find out more.

Files with missing lines Coverage Δ
onedal/_device_offload.py 86.74% <61.53%> (-3.26%) ⬇️

... and 3 files with indirect coverage changes

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

@maria-Petrova maria-Petrova merged commit 4a46108 into rls/2026.0.0-rls Apr 8, 2026
22 of 24 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants