jax
6fb32d3c - [mgpu] Infer memory space in `ptr_as_memref` and check consistency.

Commit

11 days ago

[mgpu] Infer memory space in `ptr_as_memref` and check consistency. PiperOrigin-RevId: 911106938

References

Author

chr1sj0nes

Committer

Google-ML-Automation

Parents