`f8e5m2` conversions on Blackwell+ (`sm_100`+).

Commit

7 days ago

[Mosaic GPU] Guard `bf16` <-> `f8e4m3fn`/`f8e5m2` conversions on Blackwell+ (`sm_100`+). `cvt.{e4m3x2,e5m2x2}.bf16x2` and `cvt.bf16x2.{e4m3x2,e5m2x2}` are not supported pre-Blackwell (`arch.major < 10`). PiperOrigin-RevId: 1004128476

References

#41703 - [DO NOT MERGE][ROCm] Migrate Pytest ROCm lane to rocmless container + pip-installed TheRock

#41863 - [Mosaic GPU] Guard `bf16` <-> `f8e4m3fn`/`f8e5m2` conversions on Blackwell+ (`sm_100`+).

#41881 - [ROCm] Enable remote compilation for jax in rocm rbe

#42067 - [ROCm] Enable remote compilation for jax

Author

Google-ML-Automation

Committer

Google-ML-Automation

Parents

d330fa6d

jax 62045164 - [Mosaic GPU] Guard `bf16` <-> `f8e4m3fn`/`f8e5m2` conversions on Blackwell+ (`sm_100`+).

jax
62045164 - [Mosaic GPU] Guard `bf16` <-> `f8e4m3fn`/`f8e5m2` conversions on Blackwell+ (`sm_100`+).