Skip to content

[TEST] Cover Gluon Blackwell tcgen05 multicast ops#464

Draft
Jokeren wants to merge 1 commit into
test/gluon-blackwell-tcgen05-two-cta-opsfrom
test/gluon-blackwell-tcgen05-multicast-ops
Draft

[TEST] Cover Gluon Blackwell tcgen05 multicast ops#464
Jokeren wants to merge 1 commit into
test/gluon-blackwell-tcgen05-two-cta-opsfrom
test/gluon-blackwell-tcgen05-multicast-ops

Conversation

@Jokeren

@Jokeren Jokeren commented Jun 20, 2026

Copy link
Copy Markdown
Member

Summary

Add CPU-result coverage for Blackwell TMA multicast feeding tcgen05_mma through Gluon.

This adds a focused end-to-end test that:

  • launches with num_ctas=4
  • loads A/B tiles with tma.async_load(..., multicast=True)
  • runs two blackwell.tcgen05_mma(..., multicast=True) steps into a two-CTA tensor-memory accumulator
  • exercises accumulation with use_acc=True on the second K tile
  • stores the result back through TMA
  • checks the output against a PyTorch CPU matmul reference

Validation

  • PYTHONPATH=/mnt/keren/triton-viz pytest tests/end_to_end/test_gluon_blackwell_tcgen05_multicast_ops.py -q
  • PYTHONPATH=/mnt/keren/triton-viz pytest tests/end_to_end/test_gluon_blackwell_tcgen05_multicast_ops.py tests/end_to_end/test_gluon_blackwell_tcgen05_two_cta_ops.py tests/end_to_end/test_gluon_blackwell_tensor_memory_reduction_ops.py tests/end_to_end/test_gluon_blackwell_tcgen05_copy_ops.py tests/end_to_end/test_gluon_blackwell_tcgen05_scaled_ops.py tests/end_to_end/test_gluon_blackwell_tcgen05_ops.py tests/end_to_end/test_gluon_blackwell_tensor_memory_ops.py tests/end_to_end/test_gluon_wgmma_ops.py tests/end_to_end/test_gluon_tma_im2col_ops.py tests/end_to_end/test_gluon_blackwell_tma_ops.py tests/end_to_end/test_gluon_tma_ops.py tests/end_to_end/test_gluon_async_copy_ops.py tests/end_to_end/test_gluon_core_ops.py tests/end_to_end/test_gluon.py tests/end_to_end/test_core.py tests/unit/test_adapters.py -q
  • pre-commit run --files tests/end_to_end/test_gluon_blackwell_tcgen05_multicast_ops.py

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