PR #615 Enable test_dot_general_sharded for ROCm

Enable test_dot_general_sharded for ROCm #615

gulsumgudukbay merged 1 commit into rocm-jaxlib-v0.8.0 from phambinh/enable-scaled-dot-sharded-rocm

Enable test_dot_general_sharded for ROCm

794f5623

phambinhfin requested a review 2 days ago

gulsumgudukbay approved these changes on 2026-01-20

phambinhfin added open-upstream

phambinhfin added cherry-pick-candidate

gulsumgudukbay merged 228310bd into rocm-jaxlib-v0.8.0 2 days ago

gulsumgudukbay deleted the phambinh/enable-scaled-dot-sharded-rocm branch 2 days ago

Reviewers

gulsumgudukbay

Assignees

No one assigned

Labels

open-upstream cherry-pick-candidate

Milestone

No milestone