Skip to content

[core] feat: improve variant resolution#330

Open
sayakpaul wants to merge 5 commits intomainfrom
improve-variant-resolution
Open

[core] feat: improve variant resolution#330
sayakpaul wants to merge 5 commits intomainfrom
improve-variant-resolution

Conversation

@sayakpaul
Copy link
Member

@sayakpaul sayakpaul commented Mar 11, 2026

Fixes #326

Should we document it somewhere?

@sayakpaul sayakpaul requested a review from danieldk March 11, 2026 05:01
@HuggingFaceDocBuilderDev

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.

danieldk
danieldk previously approved these changes Mar 11, 2026
Copy link
Member

@danieldk danieldk left a comment

Choose a reason for hiding this comment

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

Looks great, one tiny minor nit. Feel free to merge!


def test_xpu_returns_single_variant(self):
backend = XPU(version=Version("2024.2"))
assert _compatible_backend_variants(backend) == [backend.variant]
Copy link
Member

Choose a reason for hiding this comment

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

Probably best to put a literal here as well on the RHS of the equality check?

@sayakpaul
Copy link
Member Author

@danieldk I updated to using literal. But I need another approval for merging.

@sayakpaul sayakpaul requested a review from danieldk March 11, 2026 10:17
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.

Support resolving the CUDA version to another compatible version

3 participants