jax
0ee9531e - [Pallas:MGPU] Add support for indexed refs to WGMMA

Commit

1 year ago

[Pallas:MGPU] Add support for indexed refs to WGMMA PiperOrigin-RevId: 687258992

References

#24363 - [Pallas:MGPU] Add support for indexed refs to WGMMA

Author

apaszke

apaszke

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading