jax
Add proper ROCm skip for cuDNN attention tests
#604

Merged

Add proper ROCm skip for cuDNN attention tests #604

gulsumgudukbay merged 1 commit into rocm-jaxlib-v0.8.0 from phambinh/nn-cudnn-rocm-skip

Add proper ROCm skip for cuDNN attention tests

06c87dd9

phambinhfin requested a review 153 days ago

Arech8 commented on 2026-01-16

gulsumgudukbay requested a review from

Arech8 148 days ago

gulsumgudukbay approved these changes on 2026-01-21

gulsumgudukbay added open-upstream

gulsumgudukbay added cherry-pick-candidate

gulsumgudukbay merged ad0ffd93 into rocm-jaxlib-v0.8.0 148 days ago

gulsumgudukbay deleted the phambinh/nn-cudnn-rocm-skip branch 148 days ago

Reviewers

gulsumgudukbay

Arech8

Assignees

No one assigned

Labels

open-upstream cherry-pick-candidate

Milestone

No milestone