Skip to content

handle pytorch nightly version strings#101

Open
druttka wants to merge 1 commit intosczhou:mainfrom
druttka:druttka/is-high-version-for-pytorch-nightlies
Open

handle pytorch nightly version strings#101
druttka wants to merge 1 commit intosczhou:mainfrom
druttka:druttka/is-high-version-for-pytorch-nightlies

Conversation

@druttka
Copy link
Copy Markdown

@druttka druttka commented Apr 16, 2025

When using pytorch nightlies for newer versions of cuda, the regex fails to match. Since IS_HIGH_VERSION is only used for conditionals to guard checking backends.mps, this PR replaces the regex with a hasattr check for mps, and renames to IS_MPS_AWARE for clarity.

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.

1 participant