Skip to content

Add conditional replacement of @torch.inference_mode for inference on AMD DirectML GPUs #3295

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.

Sign up for GitHub

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

Merged
deruyter92 merged 9 commits into from
May 12, 2026
Merged

Add conditional replacement of @torch.inference_mode for inference on AMD DirectML GPUs #3295

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
af2a079
Add conditional no_grad decorator for AMD DirectML GPUs.
deruyter92 Apr 28, 2026
5c52ede
Update comment
deruyter92 Apr 28, 2026
6ae4829
update naming: `_no_grad_decorator` -> `_inference_mode_decorator`
deruyter92 Apr 28, 2026
880394b
Add pytest for directml conditional no_grad inference mode
deruyter92 Apr 28, 2026
03ac5a0
update pyproject.toml and uv.lock: add importlib to `dev` dependencies
deruyter92 Apr 28, 2026
199501c
Update deeplabcut/pose_estimation_pytorch/runners/inference.py
deruyter92 Apr 28, 2026
a4d29d8
restore pyproject.toml and uv.lock (no changes needed)
deruyter92 Apr 28, 2026
4c1bdb1
add error-hint (using contextmanager) for directml-related runtime error
deruyter92 Apr 28, 2026
662555f
add note in TechHardware.md for DirectML inference troubleshooting
deruyter92 Apr 28, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
update pyproject.toml and uv.lock: add importlib to dev dependencies
  • Loading branch information
@deruyter92
deruyter92 committed Apr 28, 2026
commit 03ac5a0efc00d7cc565236259841d610f3935303
1 change: 1 addition & 0 deletions pyproject.toml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -113,6 +113,7 @@ Documentation = "https://deeplabcut.github.io/DeepLabCut/README.html"
[dependency-groups]
dev = [
"coverage",
"importlib",
Comment thread
deruyter92 marked this conversation as resolved.
Outdated
"nbformat>5",
"pre-commit",
"pytest",
Expand Down
Loading
Loading