Skip to content
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.

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

[benchmarks] Fix CUDA device check. #6419

Merged
merged 1 commit into from
Jan 31, 2024
Merged

Conversation

ysiraichi
Copy link
Collaborator

This PR removes the check we had for CUDA:

def _is_available(self, ...):
    ...
    if cfg_accelerator == "cuda":
        if cfg_xla is None and not is_xla_device_available("CUDA"):
            return False
    ...

@miladm

@ysiraichi ysiraichi merged commit be44bc6 into master Jan 31, 2024
18 checks passed
amithrm pushed a commit to amithrm/xla that referenced this pull request Mar 1, 2024
bhavya01 pushed a commit that referenced this pull request Apr 22, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants