llvm-project
635b4166 - [MLIR][NVVM][NVPTX] Add collector_b operand to tcgen05.mma intrinsics (#214280)

Commit
41 days ago
[MLIR][NVVM][NVPTX] Add collector_b operand to tcgen05.mma intrinsics (#214280) Extend `tcgen05.mma`, `disable_output_lane`, and `block_scale` NVVM intrinsics with a `collector_b` immediate operand. Update the NVPTX patterns to select `collector::b` forms when the subtarget supports them, and enable the relevant sm_107f PTX 9.4 predicate. Auto-upgrade older IR by appending `collector_usage_b=discard` so existing bitcode keeps the previous behavior. Extend tcgen05.mma MLIR Ops with `collectorOpB` argument.
Author
Parents
Loading