jax
283e4c77 - register rocm to block_scaled_dot lowering path

Commit

58 days ago

register rocm to block_scaled_dot lowering path

References

#33757 - register rocm to block_scaled_dot lowering path

Author

ScXfjiang

ScXfjiang

Parents

Loading