jax
Enable testDotProductAttentionBiasGradient on ROCm
#637

Merged

Enable testDotProductAttentionBiasGradient on ROCm #637

phambinhfin merged 1 commit into rocm-jaxlib-v0.8.0 from phambinh/unskip-attention-bias-gradient-rocm

Enable testDotProductAttentionBiasGradient on ROCm

054db7a8

phambinhfin requested a review 155 days ago

magaonka-amd added open-upstream

magaonka-amd added cherry-pick-candidate

magaonka-amd approved these changes on 2026-01-22

phambinhfin merged fa0512a2 into rocm-jaxlib-v0.8.0 155 days ago

Reviewers

magaonka-amd

Assignees

No one assigned

Labels

open-upstream cherry-pick-candidate

Milestone

No milestone