master

文件	最后提交记录	最后更新时间
CMakeLists.txt	[mlir][nvgpu] NFC - move NVGPU conversion helpers to NvGpu utils library The ConvertVectorToGpu pass implementation contained a small private support library for performing various calculations during conversion between vector and nvgpu.mma.sync and nvgpu.ldmatrix operations. The support library is moved under Dialect/NVGPU/Utils because the functions have wider utility. Some documentation comments are added or improved. Reviewed By: ThomasRaoux Differential Revision: https://reviews.llvm.org/D135303	3 年前
MMAUtils.cpp	[mlir][VectorToGPU] Update memref stride preconditions on nvgpu.mma.sync path This change removes the requirement that the row stride be statically known when converting vector.transfer_read and vector.transfer_write to distributed SIMT operations in the nvgpu lowering path. It also adds a check to verify that the last dimension of the source memref is statically known to have stride 1 since this is assumed in the conversion logic. No other change should be required since the generated vector.load operations are never created across dimensions other than the last. The routines for checking preconditions on vector.transfer_read/write are moved to under nvgpu utilities. The change is NFC with respect to the GPU dialect lowering path. Reviewed By: ThomasRaoux Differential Revision: https://reviews.llvm.org/D155753	2 年前