jax
[ROCm] Skip 6 Pallas FusedAttentionTest variants exceeding gfx942 LDS limit
#797

Open

[ROCm] Skip 6 Pallas FusedAttentionTest variants exceeding gfx942 LDS limit #797

srinivamd wants to merge 4 commits into rocm-jaxlib-v0.9.1 from skip-fused-attention-lds-v0.9.1

[ROCm] Skip 6 Pallas FusedAttentionTest variants exceeding gfx942 LDS…

ba850bae

fix: restore missing -m filter and multi-GPU block after snippet error

88fa29c3

fix: restore multi-GPU pytest block corrupted by prior snippet

c21ec7a2

fix: restore second_cmd_retval and else/fi for multi-GPU block

dd1c9a10

srinivamd requested a review from

charleshofer 18 days ago

srinivamd requested a review from

Ruturaj4 18 days ago

srinivamd requested a review from

Shahzeb-AMD 18 days ago

Reviewers

charleshofer

Ruturaj4

Shahzeb-AMD

Assignees

No one assigned

Labels

None yet

Milestone

No milestone