[mlir][nvgpu] Add `mbarrier.arrive.expect_tx` and `mbarrier.try_wait.parity`
This work adds two Ops: `mbarrier.arrive.expect_tx` performs expect_tx `mbarrier.barrier` returns `mbarrier.barrier.token` `mbarrier.try_wait.parity` waits on `mbarrier.barrier` and `mbarrier.barrier.token` `mbarrier.arrive.expect_tx` is one of the requirement to enable H100 TMA support. Depends on D154074 D154076 D154059 D154060 Reviewed By: qcolombet Differential Revision: https://reviews.llvm.org/D154094
Loading
Please sign in to comment