Cherry-pick: [ROCm] Fix bf16 upcast handling for libdevice calls.#793

Open

magaonka-amd wants to merge 1 commit intoROCm:rocm-jaxlib-v0.9.2from

magaonka-amd:fix/bf16-libdevice-upcast-v092

Commits on Apr 12, 2026

PR #39871: [ROCm] Fix bf16 upcast handling for libdevice calls.

zoranjovanovic-ns
authored and
magaonka-amd
committed