Searched refs:matB (Results 1 – 2 of 2) sorted by relevance
| /llvm-project-15.0.7/mlir/test/Dialect/NVGPU/ |
| H A D | optimize-shared-memory.mlir | 51 %matB = nvgpu.ldmatrix %shmB[%fragRow, %fragCol] {numTiles = 4 : i32, transpose = false}
54 return %mat, %matB: vector<4x2xf16>, vector<4x2xf16>
144 %matB = nvgpu.ldmatrix %shmB[%fragRow, %fragCol] {numTiles = 4 : i32, transpose = false}
155 …return %mat, %matB, %elem, %elem2, %elemB: vector<4x1xf32>, vector<4x1xf32>, f32, vector<4xf32>, f…
|
| /llvm-project-15.0.7/mlir/lib/Conversion/NVGPUToNVVM/ |
| H A D | NVGPUToNVVM.cpp | 307 SmallVector<Value> matB = in matchAndRewrite() local
316 op.getLoc(), intrinsicResTy, matA, matB, matC, in matchAndRewrite()
|