Skip to content

test(sdpa): fuzz head-axis stride gaps in the ragged sweeps #540

Description

@vedaanta

Follow-up to #516/#526. The with_ragged_token_gap knob fuzzes per-tensor TOKEN-stride gaps (nhd tokens). Head-axis gaps (stride[1] != d, e.g. head-interleaved buffers) are expressible by the native path in #526 (the TMA descriptors and Q/O offset math read stride[1] too) but are not fuzzed — extend the knob or add a sibling so the L0 ragged sweeps cover them, with gaps constrained to the 16-byte granularity rule.

🤖 Generated with Claude Code

Metadata

Metadata

Assignees

No one assigned

    Labels

    cat-enhancementsmod-infraInfrastructure, CI/CD, build systems, packaging, releases, or repo maintenance.orig-nv-engReported or requested by NVIDIA engineering.

    Type

    No type

    Projects

    No projects

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions