Skip to content

feat(lower-tirx): support PTX movmatrix - #20171

Merged
spectrometerHBH merged 2 commits into
apache:mainfrom
spectrometerHBH:cudnn-frontend
Aug 24, 2026
Merged

spectrometerHBH merged 2 commits into
apache:mainfrom
spectrometerHBH:cudnn-frontend

Conversation

@spectrometerHBH

@spectrometerHBH spectrometerHBH commented Aug 24, 2026 •

Copy link
Copy Markdown
Contributor

Summary

  • register movmatrix.sync.aligned.m8n8.trans.b16 in the CUDA PTX instruction table
  • cover the emitted mnemonic in the extracted intrinsic codegen test
  • regenerate the checked-in TIRx PTX stub and update the certified variant count

Testing

  • python -m pytest -q tests/python/tirx/codegen/test_ptx_dialect.py (47 passed, 32 skipped)
  • python -m pytest -q tests/python/tirx/codegen/test_codegen_cuda.py -k megamoe_extracted_intrinsics_codegen (1 passed)
  • pre-commit run --files python/tvm/backend/cuda/ptx/table.py python/tvm/script/tirx.pyi tests/python/tirx/codegen/test_codegen_cuda.py tests/python/tirx/codegen/test_ptx_dialect.py

@spectrometerHBH
spectrometerHBH merged commit a29a5e9 into apache:main Aug 24, 2026
8 checks passed
tqchen pushed a commit that referenced this pull request Oct 16, 2026
## Summary

- register `movmatrix.sync.aligned.m8n8.trans.b16` in the CUDA PTX
instruction table
- cover the emitted mnemonic in the extracted intrinsic codegen test
- regenerate the checked-in TIRx PTX stub and update the certified
variant count

## Testing

- `python -m pytest -q tests/python/tirx/codegen/test_ptx_dialect.py`
(47 passed, 32 skipped)
- `python -m pytest -q tests/python/tirx/codegen/test_codegen_cuda.py -k
megamoe_extracted_intrinsics_codegen` (1 passed)
- `pre-commit run --files python/tvm/backend/cuda/ptx/table.py
python/tvm/script/tirx.pyi
tests/python/tirx/codegen/test_codegen_cuda.py
tests/python/tirx/codegen/test_ptx_dialect.py`
tqchen pushed a commit that referenced this pull request Oct 18, 2026
## Summary

- register `movmatrix.sync.aligned.m8n8.trans.b16` in the CUDA PTX
instruction table
- cover the emitted mnemonic in the extracted intrinsic codegen test
- regenerate the checked-in TIRx PTX stub and update the certified
variant count

## Testing

- `python -m pytest -q tests/python/tirx/codegen/test_ptx_dialect.py`
(47 passed, 32 skipped)
- `python -m pytest -q tests/python/tirx/codegen/test_codegen_cuda.py -k
megamoe_extracted_intrinsics_codegen` (1 passed)
- `pre-commit run --files python/tvm/backend/cuda/ptx/table.py
python/tvm/script/tirx.pyi
tests/python/tirx/codegen/test_codegen_cuda.py
tests/python/tirx/codegen/test_ptx_dialect.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.

2 participants