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

[ROCm]: Add get_arch_details for triton kernel call #22667

Open

rahulbatra85 wants to merge 1 commit into google:main from ROCm:rocm-jax-triton-add-get_arch_detail

+13 −0

Contributor

rahulbatra85 commented Jul 25, 2024 •

edited

Loading

Creating a Triton backend object in JAX-Triton requires arch detail for ROCM. This adds the API on the JAX side


[ROCm]: Add get_arch_details for triton kernel call

6325abe

Contributor Author

rahulbatra85 commented Jul 25, 2024

@superbobry @hawkinsp Please review. Thanks!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment