jax
c7d535d3 - [pallas_mgpu] Add a test for emit_pipeline with wgmma.

Commit

1 year ago

[pallas_mgpu] Add a test for emit_pipeline with wgmma. PiperOrigin-RevId: 723012611

References

Author

petebu

Committer

Google-ML-Automation

Parents