Skip to content

Add XPU support for Torch 2.11#409

Merged
danieldk merged 8 commits intomainfrom
pytorch-2.11-xpu
Mar 27, 2026
Merged

Add XPU support for Torch 2.11#409
danieldk merged 8 commits intomainfrom
pytorch-2.11-xpu

Conversation

@danieldk
Copy link
Copy Markdown
Member

Changes:

  • Update RHEL to Nix script to support multiple versions of the same package names.
  • Add oneAPI 2025.3.2.
  • Add Torch 2.11 XPU.
  • Build triton-xpu with binary wheels, replacing outdated versions.
  • Fix issue with wheel dependency relaxation hook (e.g. triton-xpu getting relaxed to triton).
  • Remove old, now-unused oneAPI versions.

Also add oneAPI 2025.3.2 as a requirement for Torch 2.11.
If a package like `triton-xpu` is required and the wheel `triton`
should be relaxed, `triton-xpu` got shortened to `triton` due to
too greedy matching in the hook. This change requires the first
character in the version specifier to be a non-PEP 508 character
to avoid this.
Also update to 3.6.0 and 3.7.0 for Torch 2.10 and Torch 2.11
respectively
@HuggingFaceDocBuilderDev
Copy link
Copy Markdown

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

"nvidia-cufile"
];
suffix = [
""
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

is this empty suffix used? just looks a bit strange

Copy link
Copy Markdown
Member Author

Choose a reason for hiding this comment

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

Yep, in some cases the dependencies do not have a -cuNN suffix. But it's really weird, because the package might have it for CUDA 12 and not for CUDA 13.

Copy link
Copy Markdown
Collaborator

@drbh drbh left a comment

Choose a reason for hiding this comment

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

lgtm, just added a small question/comment above

@danieldk danieldk merged commit ba07ad3 into main Mar 27, 2026
39 checks passed
@danieldk danieldk mentioned this pull request Mar 30, 2026
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.

3 participants