TThomas Raoux[mlir][NVGPU] Verifier for nvgpu.ldmatrix
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
[mlir][NVGPU] Verifier for nvgpu.ldmatrix * Adds verifiers for nvgpu.ldmatrix op * Adds tests to mlir/test/Dialect/NVGPU/invalid.mlir Reviewed By: ThomasRaoux Differential Revision: https://reviews.llvm.org/D129669 | 3 年前 | |
[mlir] Flip accessors to prefixed form (NFC) Another mechanical sweep to keep diff small for flip to _Prefixed. | 3 年前 | |
[mlir][nvgpu] shared memory access optimization pass This change adds a transformation and pass to the NvGPU dialect that attempts to optimize reads/writes from a memref representing GPU shared memory in order to avoid bank conflicts. Given a value representing a shared memory memref, it traverses all reads/writes within the parent op and, subject to suitable conditions, rewrites all last dimension index values such that element locations in the final (col) dimension are given by newColIdx = col % vecSize + perm[row](col/vecSize,row) where perm is a permutation function indexed by row and vecSize is the vector access size in elements (currently assumes 128bit vectorized accesses, but this can be made a parameter). This specific transformation can help optimize typical distributed & vectorized accesses common to loading matrix multiplication operands to/from shared memory. Differential Revision: https://reviews.llvm.org/D127457 | 3 年前 |