Skip to content

[TEST] Cover Gluon Blackwell tcgen05 copy ops#461

Draft
Jokeren wants to merge 11 commits into
test/gluon-blackwell-tcgen05-scaled-opsfrom
test/gluon-blackwell-tcgen05-copy-ops
Draft

[TEST] Cover Gluon Blackwell tcgen05 copy ops#461
Jokeren wants to merge 11 commits into
test/gluon-blackwell-tcgen05-scaled-opsfrom
test/gluon-blackwell-tcgen05-copy-ops

Conversation

@Jokeren

@Jokeren Jokeren commented Jun 20, 2026

Copy link
Copy Markdown
Member

Summary

Add CPU-result coverage for Blackwell tcgen05_copy simulation through Gluon.

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

  • loads a strided 64x64 tile from global memory
  • stores it in shared memory
  • copies shared memory into tensor memory with blackwell.tcgen05_copy
  • commits/waits on the tcgen05 operation
  • loads tensor memory back and stores into a strided output view
  • checks the output view against the CPU input tile

Validation

  • PYTHONPATH=/mnt/keren/triton-viz pytest tests/end_to_end/test_gluon_blackwell_tcgen05_copy_ops.py -q
  • PYTHONPATH=/mnt/keren/triton-viz pytest 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_copy_ops.py

@Jokeren Jokeren force-pushed the test/gluon-blackwell-tcgen05-scaled-ops branch 2 times, most recently from 59256be to b05edb7 Compare June 26, 2026 17:06
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