Skip to content

feat(hip/gfx1201): RDNA4 DS4 WMMA12 index score, IQ dequant, and MM3 kernel uplifts - #984

Draft
nicholas631 wants to merge 2 commits into
mw00:mainfrom
nicholas631:feat/hip-gfx1201-rdna4-wmma12-row-local-v1163
Draft

nicholas631 wants to merge 2 commits into
mw00:mainfrom
nicholas631:feat/hip-gfx1201-rdna4-wmma12-row-local-v1163

Conversation

@nicholas631

Copy link
Copy Markdown
Contributor

Summary

Clean Draft PR rebased onto Project Maya v1.1.63 (6651b5b) targeting AMD RDNA4 (gfx1201) and shared HIP/MM3 kernel parity, with zero host-specific recipes or benchmark artifacts.

Modified Files

  • src/ds4/ds4_kernels.cu: RDNA4 gfx12 WMMA12 index score (DS4_INDEX_TC=1) uplift.
  • src/ds4/ds4_iq.cu: RDNA4 wave32 / HIP IQ dequantization uplift.
  • src/kernels/cuda/iq_kernels.cu: Allow Q8_0 down-projection in native_expert_grouped.
  • src/mm3/mm3_kernels.cu, src/mm3/mm3_kernels.cuh, src/mm3/mm3_gguf.cu, tests/mm3_gguf_test.cu: MM3 RDNA4 kernel dispatch and Q8_0 GGUF parity.

Validation & Numerical Parity

  • Reproducer: ./build-hip/mm3_kernels_test, ./build-hip/mm3_gguf_test, ./build-hip/iq_mmvq_parity_test, ./build-hip/glm_attn_bench --parity-only
  • Numerical parity: mm3_kernels_test, mm3_gguf_test, iq_mmvq_parity_test, suffix_drafter_test, mtp_rows_test all PASS (max_abs = 0.000 / within FP16 WMMA tolerance).
  • Non-target architectures (sm_86, gfx906) unaffected.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant