jax
c95381d5 - [ROCm]: call rocm function when lowering sparse ops

Commit

2 years ago

[ROCm]: call rocm function when lowering sparse ops

References

rocm-jaxlib-v0.4.20-w-sparse-fix

Author

Rahul Batra

Parents

Loading