colin
/
onnx-mlir
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Normalize memrefs with affine_map in krnl.getref (#327) 

Added a trait of `MemRefsNormalizable` in `krnl.getref` to have
`krnl.getref` normalizable.
Reference: https://github.com/onnx/onnx-mlir/pull/322
           https://reviews.llvm.org/D86236

Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
This commit is contained in:
Haruki Imai 2020-10-10 05:59:52 +09:00 committed by GitHub
parent 7bfb5c93c1
commit 533d47acf1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/Dialect/Krnl/KrnlOps.td
View File

@ -218,7 +218,7 @@ def KrnlPackedConstantOp : Op<Krnl_Dialect, "packed_const"> {
let printer = ?; let printer = ?;
} }
def KrnlGetRefOp : Op<Krnl_Dialect, "getref"> { def KrnlGetRefOp : Op<Krnl_Dialect, "getref", [MemRefsNormalizable]> {
let summary = "Krnl a MemRef from within another MemRef starting at a specific offset."; let summary = "Krnl a MemRef from within another MemRef starting at a specific offset.";
let description = [{ let description = [{
Retreieves a MemRef from within another MemRef: Retreieves a MemRef from within another MemRef:

14
test/mlir/krnl/krnl_normalization.mlir
View File

@ -14,3 +14,17 @@ func @test_krnl_memcpy_norm(%arg0: memref<1x16384xf32>) -> memref<1x16x4x4xf32,
return %0 : memref<1x16x4x4xf32, #map_tile> return %0 : memref<1x16x4x4xf32, #map_tile>
// CHECK: return [[ALLOC]] : memref<1x16x1x1x32x32xf32> // CHECK: return [[ALLOC]] : memref<1x16x1x1x32x32xf32>
} }
// CHECK-LABEL: test_getref_norm
func @test_getref_norm() -> () {
%c0_i64 = constant 0 : i64
%0 = alloc() : memref<1x81920xf32>
%1 = alloc() : memref<1x16x4x4xf32, #map_tile>
// CHECK: [[ALLOC:%.+]] = alloc() : memref<1x16x1x1x32x32xf32>
%2 = "krnl.getref"(%0, %c0_i64) : (memref<1x81920xf32>, i64) -> memref<1x16x4x4xf32>
// Do something using %1 and %2
dealloc %1: memref<1x16x4x4xf32, #map_tile>
// CHECK: dealloc [[ALLOC:%.+]] : memref<1x16x1x1x32x32xf32>
dealloc %0: memref<1x81920xf32>
return
}