jax
76ccb199 - [pallas:mosaic_gpu] Added some runtime type checking to `copy_` and `barrier_` primitives

Commit

1 year ago

[pallas:mosaic_gpu] Added some runtime type checking to `copy_*` and `barrier_*` primitives PiperOrigin-RevId: 710302436

References

Author

superbobry

Committer

Google-ML-Automation

Parents

jax 76ccb199 - [pallas:mosaic_gpu] Added some runtime type checking to `copy_*` and `barrier_*` primitives