Skip to content

enable mps pytorch backend if available and some housekeeping#481

Merged
FynnBe merged 13 commits intomainfrom
dev
Apr 15, 2026
Merged

enable mps pytorch backend if available and some housekeeping#481
FynnBe merged 13 commits intomainfrom
dev

Conversation

@FynnBe
Copy link
Copy Markdown
Member

@FynnBe FynnBe commented Apr 1, 2026

No description provided.

@FynnBe FynnBe requested a review from k-dominik April 1, 2026 12:49
@FynnBe
Copy link
Copy Markdown
Member Author

FynnBe commented Apr 1, 2026

Closes #477

Copy link
Copy Markdown
Member

@k-dominik k-dominik left a comment

Choose a reason for hiding this comment

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

lgtm. Note that it seems that in pytorch 3.11 (not available yet) A LOT of the stuff seems now implemented - having said that I don't think it hurts to set this env variable anyway as it helps the old versions to at least produce an output rather than failing!

No clue about the early stopping you want to do there ;)

Comment thread src/bioimageio/core/_settings.py
@FynnBe FynnBe changed the title set PYTORCH_ENABLE_MPS_FALLBACK and some housekeeping enable mps pytorch backend if available and some housekeeping Apr 2, 2026
@github-actions
Copy link
Copy Markdown
Contributor

github-actions bot commented Apr 15, 2026

☂️ Python Coverage

current status: ❌

Overall Coverage

Lines Covered Coverage Threshold Status
4103 3128 76% 70% 🟢

New Files

File Coverage Status
src/bioimageio/core/_collection.py 0% 🔴
TOTAL 0% 🔴

Modified Files

File Coverage Status
src/bioimageio/core/_init_.py 100% 🟢
src/bioimageio/core/_resource_tests.py 73% 🟢
src/bioimageio/core/_settings.py 88% 🟢
src/bioimageio/core/backends/pytorch_backend.py 79% 🟢
src/bioimageio/core/backends/torchscript_backend.py 97% 🟢
src/bioimageio/core/digest_spec.py 85% 🟢
src/bioimageio/core/io.py 84% 🟢
src/bioimageio/core/sample.py 93% 🟢
src/bioimageio/core/tensor.py 83% 🟢
TOTAL 87% 🟢

updated for commit: 174b5b4 by action🐍

@FynnBe FynnBe merged commit 71a5ec1 into main Apr 15, 2026
9 checks passed
@FynnBe FynnBe deleted the dev branch April 15, 2026 09:45
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