replica
/
hanchenye-scalehls
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[Samples] Update all onnx-mlir samples

This commit is contained in:
Hanchen Ye 2021-11-12 02:32:40 -06:00
parent 22e59877cb
commit 36d5083315
7 changed files with 570 additions and 570 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
lib/Conversion/LegalizeOnnx.cpp
View File

@ -91,9 +91,9 @@ void LegalizeOnnx::runOnOperation() {
for (auto index : loadOp.getIndices()) {
// Handle constant defining operation.
if (auto defOp = index.getDefiningOp())
if (auto constOp = dyn_cast<ConstantOp>(defOp))
if (auto constOp = dyn_cast<arith::ConstantOp>(defOp))
if (constOp.getType().isa<IndexType>())
if (auto constAttr = constOp.getValue().dyn_cast<IntegerAttr>()) {
if (auto constAttr = constOp.value().dyn_cast<IntegerAttr>()) {
exprs.push_back(
builder.getAffineConstantExpr(constAttr.getUInt()));
continue;
@ -136,7 +136,7 @@ void LegalizeOnnx::runOnOperation() {
// If the kernel global operation gets a value, create a standard
// constant operation to substitute it.
builder.setInsertionPoint(&op);
auto tensor = builder.create<ConstantOp>(op.getLoc(), value);
auto tensor = builder.create<arith::ConstantOp>(op.getLoc(), value);
auto memref = builder.create<memref::BufferCastOp>(
op.getLoc(), op.getResult(0).getType(), tensor);
op.getResult(0).replaceAllUsesWith(memref);

4
lib/Transforms/Graph/LegalizeDataflow.cpp
View File

@ -42,8 +42,8 @@ static void getSuccessorsMap(Block &block, SuccessorsMap &map) {
// traverse will not enter any control flow operations.
for (auto &op : block.getOperations()) {
// TODO: Some operations are dataflow source, which will not be scheduled.
if (isa<memref::AllocOp, memref::AllocaOp, ConstantOp, memref::TensorLoadOp,
memref::BufferCastOp>(op))
if (isa<memref::AllocOp, memref::AllocaOp, arith::ConstantOp,
memref::TensorLoadOp, memref::BufferCastOp>(op))
continue;
// Collect all memref results if the current operation is a loop.

70
samples/onnx-mlir/lenet/lenet.mlir
View File

@ -17,10 +17,10 @@
module {
%0 = "krnl.packed_const"() {file_name = "/tmp/packed_const-b6b002.tmp", is_le = true, size_in_bytes = 247896 : i64} : () -> i64
func @main_graph(%arg0: memref<1x3x32x32xf32>) -> memref<1x10xf32> attributes {input_names = ["input.1"], output_names = ["18"]} {
%c16 = constant 16 : index
%c5 = constant 5 : index
%cst = constant 1.000000e+00 : f32
%cst_0 = constant 0.000000e+00 : f32
%c16 = arith.constant 16 : index
%c5 = arith.constant 5 : index
%cst = arith.constant 1.000000e+00 : f32
%cst_0 = arith.constant 0.000000e+00 : f32
%1 = memref.alloc() : memref<1x10xf32>
%2 = memref.alloc() : memref<1x84xf32>
%3 = memref.alloc() : memref<1x84xf32>
@ -31,7 +31,7 @@ module {
%8 = memref.alloc() : memref<1x16x5x5xf32>
%9 = memref.alloc() : memref<1x6x14x14xf32>
%10 = memref.alloc() : memref<1x6x14x14xf32>
%11 = "krnl.global"() {name = "constant_0", offset = 0 : i64, shape = [6, 3, 5, 5]} : () -> memref<6x3x5x5xf32>
%11 = "krnl.global"() {name = "arith.constant_0", offset = 0 : i64, shape = [6, 3, 5, 5]} : () -> memref<6x3x5x5xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 6 {
affine.for %arg3 = 0 to 14 {
@ -45,8 +45,8 @@ module {
%21 = affine.load %arg0[%arg1, %arg5, %19, %20] : memref<1x3x32x32xf32>
%22 = affine.load %11[%arg2, %arg5, %arg6, %arg7] : memref<6x3x5x5xf32>
%23 = affine.load %10[%arg1, %arg2, %arg3, %arg4] : memref<1x6x14x14xf32>
%24 = mulf %21, %22 : f32
%25 = addf %23, %24 : f32
%24 = arith.mulf %21, %22 : f32
%25 = arith.addf %23, %24 : f32
affine.store %25, %10[%arg1, %arg2, %arg3, %arg4] : memref<1x6x14x14xf32>
}
}
@ -60,14 +60,14 @@ module {
affine.for %arg3 = 0 to 14 {
affine.for %arg4 = 0 to 14 {
%19 = affine.load %10[%arg1, %arg2, %arg3, %arg4] : memref<1x6x14x14xf32>
%20 = cmpf "olt", %19, %cst_0 : f32
%20 = arith.cmpf "olt", %19, %cst_0 : f32
%21 = select %20, %cst_0, %19 : f32
affine.store %21, %9[%arg1, %arg2, %arg3, %arg4] : memref<1x6x14x14xf32>
}
}
}
}
%12 = "krnl.global"() {name = "constant_1", offset = 1800 : i64, shape = [16, 6, 5, 5]} : () -> memref<16x6x5x5xf32>
%12 = "krnl.global"() {name = "arith.constant_1", offset = 1800 : i64, shape = [16, 6, 5, 5]} : () -> memref<16x6x5x5xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 16 {
affine.for %arg3 = 0 to 5 {
@ -81,8 +81,8 @@ module {
%21 = affine.load %9[%arg1, %arg5, %19, %20] : memref<1x6x14x14xf32>
%22 = affine.load %12[%arg2, %arg5, %arg6, %arg7] : memref<16x6x5x5xf32>
%23 = affine.load %8[%arg1, %arg2, %arg3, %arg4] : memref<1x16x5x5xf32>
%24 = mulf %21, %22 : f32
%25 = addf %23, %24 : f32
%24 = arith.mulf %21, %22 : f32
%25 = arith.addf %23, %24 : f32
affine.store %25, %8[%arg1, %arg2, %arg3, %arg4] : memref<1x16x5x5xf32>
}
}
@ -96,7 +96,7 @@ module {
affine.for %arg3 = 0 to 5 {
affine.for %arg4 = 0 to 5 {
%19 = affine.load %8[%arg1, %arg2, %arg3, %arg4] : memref<1x16x5x5xf32>
%20 = cmpf "olt", %19, %cst_0 : f32
%20 = arith.cmpf "olt", %19, %cst_0 : f32
%21 = select %20, %cst_0, %19 : f32
affine.store %21, %7[%arg1, %arg2, %arg3, %arg4] : memref<1x16x5x5xf32>
}
@ -114,8 +114,8 @@ module {
}
}
}
%13 = "krnl.global"() {name = "constant_2", offset = 11400 : i64, shape = [120, 400]} : () -> memref<120x400xf32>
%14 = "krnl.global"() {name = "constant_3", offset = 203400 : i64, shape = [120]} : () -> memref<120xf32>
%13 = "krnl.global"() {name = "arith.constant_2", offset = 11400 : i64, shape = [120, 400]} : () -> memref<120x400xf32>
%14 = "krnl.global"() {name = "arith.constant_3", offset = 203400 : i64, shape = [120]} : () -> memref<120xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 120 {
affine.store %cst_0, %5[%arg1, %arg2] : memref<1x120xf32>
@ -123,28 +123,28 @@ module {
%24 = affine.load %6[%arg1, %arg3] : memref<1x400xf32>
%25 = affine.load %13[%arg2, %arg3] : memref<120x400xf32>
%26 = affine.load %5[%arg1, %arg2] : memref<1x120xf32>
%27 = mulf %24, %25 : f32
%28 = addf %26, %27 : f32
%27 = arith.mulf %24, %25 : f32
%28 = arith.addf %26, %27 : f32
affine.store %28, %5[%arg1, %arg2] : memref<1x120xf32>
}
%19 = affine.load %5[%arg1, %arg2] : memref<1x120xf32>
%20 = mulf %cst, %19 : f32
%20 = arith.mulf %cst, %19 : f32
%21 = affine.load %14[%arg2] : memref<120xf32>
%22 = mulf %cst, %21 : f32
%23 = addf %20, %22 : f32
%22 = arith.mulf %cst, %21 : f32
%23 = arith.addf %20, %22 : f32
affine.store %23, %5[%arg1, %arg2] : memref<1x120xf32>
}
}
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 120 {
%19 = affine.load %5[%arg1, %arg2] : memref<1x120xf32>
%20 = cmpf "olt", %19, %cst_0 : f32
%20 = arith.cmpf "olt", %19, %cst_0 : f32
%21 = select %20, %cst_0, %19 : f32
affine.store %21, %4[%arg1, %arg2] : memref<1x120xf32>
}
}
%15 = "krnl.global"() {name = "constant_4", offset = 203880 : i64, shape = [84, 120]} : () -> memref<84x120xf32>
%16 = "krnl.global"() {name = "constant_5", offset = 244200 : i64, shape = [84]} : () -> memref<84xf32>
%15 = "krnl.global"() {name = "arith.constant_4", offset = 203880 : i64, shape = [84, 120]} : () -> memref<84x120xf32>
%16 = "krnl.global"() {name = "arith.constant_5", offset = 244200 : i64, shape = [84]} : () -> memref<84xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 84 {
affine.store %cst_0, %3[%arg1, %arg2] : memref<1x84xf32>
@ -152,28 +152,28 @@ module {
%24 = affine.load %4[%arg1, %arg3] : memref<1x120xf32>
%25 = affine.load %15[%arg2, %arg3] : memref<84x120xf32>
%26 = affine.load %3[%arg1, %arg2] : memref<1x84xf32>
%27 = mulf %24, %25 : f32
%28 = addf %26, %27 : f32
%27 = arith.mulf %24, %25 : f32
%28 = arith.addf %26, %27 : f32
affine.store %28, %3[%arg1, %arg2] : memref<1x84xf32>
}
%19 = affine.load %3[%arg1, %arg2] : memref<1x84xf32>
%20 = mulf %cst, %19 : f32
%20 = arith.mulf %cst, %19 : f32
%21 = affine.load %16[%arg2] : memref<84xf32>
%22 = mulf %cst, %21 : f32
%23 = addf %20, %22 : f32
%22 = arith.mulf %cst, %21 : f32
%23 = arith.addf %20, %22 : f32
affine.store %23, %3[%arg1, %arg2] : memref<1x84xf32>
}
}
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 84 {
%19 = affine.load %3[%arg1, %arg2] : memref<1x84xf32>
%20 = cmpf "olt", %19, %cst_0 : f32
%20 = arith.cmpf "olt", %19, %cst_0 : f32
%21 = select %20, %cst_0, %19 : f32
affine.store %21, %2[%arg1, %arg2] : memref<1x84xf32>
}
}
%17 = "krnl.global"() {name = "constant_6", offset = 244536 : i64, shape = [10, 84]} : () -> memref<10x84xf32>
%18 = "krnl.global"() {name = "constant_7", offset = 247896 : i64, shape = [10], value = dense<[0.0802067816, 0.0380531251, -0.0453170575, 0.0238099229, -0.0322267264, 0.0956416651, -0.0513828322, -0.00329447933, 0.0870032906, -0.100006074]> : tensor<10xf32>} : () -> memref<10xf32>
%17 = "krnl.global"() {name = "arith.constant_6", offset = 244536 : i64, shape = [10, 84]} : () -> memref<10x84xf32>
%18 = "krnl.global"() {name = "arith.constant_7", offset = 247896 : i64, shape = [10], value = dense<[0.0802067816, 0.0380531251, -0.0453170575, 0.0238099229, -0.0322267264, 0.0956416651, -0.0513828322, -0.00329447933, 0.0870032906, -0.100006074]> : tensor<10xf32>} : () -> memref<10xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 10 {
affine.store %cst_0, %1[%arg1, %arg2] : memref<1x10xf32>
@ -181,15 +181,15 @@ module {
%24 = affine.load %2[%arg1, %arg3] : memref<1x84xf32>
%25 = affine.load %17[%arg2, %arg3] : memref<10x84xf32>
%26 = affine.load %1[%arg1, %arg2] : memref<1x10xf32>
%27 = mulf %24, %25 : f32
%28 = addf %26, %27 : f32
%27 = arith.mulf %24, %25 : f32
%28 = arith.addf %26, %27 : f32
affine.store %28, %1[%arg1, %arg2] : memref<1x10xf32>
}
%19 = affine.load %1[%arg1, %arg2] : memref<1x10xf32>
%20 = mulf %cst, %19 : f32
%20 = arith.mulf %cst, %19 : f32
%21 = affine.load %18[%arg2] : memref<10xf32>
%22 = mulf %cst, %21 : f32
%23 = addf %20, %22 : f32
%22 = arith.mulf %cst, %21 : f32
%23 = arith.addf %20, %22 : f32
affine.store %23, %1[%arg1, %arg2] : memref<1x10xf32>
}
}

248
samples/onnx-mlir/mobilenet/mobilenet.mlir
View File

@ -26,12 +26,12 @@
module {
%0 = "krnl.packed_const"() {file_name = "/tmp/packed_const-0f2051.tmp", is_le = true, size_in_bytes = 12781312 : i64} : () -> i64
func @main_graph(%arg0: memref<1x3x32x32xf32>) -> memref<1x10xf32> attributes {input_names = ["input.1"], output_names = ["86"]} {
%c0 = constant 0 : index
%c4_i64 = constant 4 : i64
%c1024 = constant 1024 : index
%c1 = constant 1 : index
%cst = constant 1.000000e+00 : f32
%cst_0 = constant 0.000000e+00 : f32
%c0 = arith.constant 0 : index
%c4_i64 = arith.constant 4 : i64
%c1024 = arith.constant 1024 : index
%c1 = arith.constant 1 : index
%cst = arith.constant 1.000000e+00 : f32
%cst_0 = arith.constant 0.000000e+00 : f32
%1 = memref.alloc() : memref<1x10xf32>
%2 = memref.alloc() : memref<1x1024xf32>
%3 = memref.alloc() : memref<1x1024x1x1xf32>
@ -103,7 +103,7 @@ module {
%69 = memref.alloc() : memref<1x32x32x32xf32>
%70 = memref.alloc() : memref<1x32x32x32xf32>
%71 = memref.alloc() : memref<1x3x34x34xf32>
%72 = "krnl.global"() {name = "constant_0", offset = 0 : i64, shape = [32, 3, 3, 3]} : () -> memref<32x3x3x3xf32>
%72 = "krnl.global"() {name = "arith.constant_0", offset = 0 : i64, shape = [32, 3, 3, 3]} : () -> memref<32x3x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 3 {
affine.for %arg3 = 0 to 34 {
@ -138,8 +138,8 @@ module {
%104 = affine.load %71[%arg1, %arg5, %102, %103] : memref<1x3x34x34xf32>
%105 = affine.load %72[%arg2, %arg5, %arg6, %arg7] : memref<32x3x3x3xf32>
%106 = affine.load %70[%arg1, %arg2, %arg3, %arg4] : memref<1x32x32x32xf32>
%107 = mulf %104, %105 : f32
%108 = addf %106, %107 : f32
%107 = arith.mulf %104, %105 : f32
%108 = arith.addf %106, %107 : f32
affine.store %108, %70[%arg1, %arg2, %arg3, %arg4] : memref<1x32x32x32xf32>
}
}
@ -153,14 +153,14 @@ module {
affine.for %arg3 = 0 to 32 {
affine.for %arg4 = 0 to 32 {
%102 = affine.load %70[%arg1, %arg2, %arg3, %arg4] : memref<1x32x32x32xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %69[%arg1, %arg2, %arg3, %arg4] : memref<1x32x32x32xf32>
}
}
}
}
%73 = "krnl.global"() {name = "constant_1", offset = 3456 : i64, shape = [32, 1, 3, 3]} : () -> memref<32x1x3x3xf32>
%73 = "krnl.global"() {name = "arith.constant_1", offset = 3456 : i64, shape = [32, 1, 3, 3]} : () -> memref<32x1x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 32 {
affine.for %arg3 = 0 to 34 {
@ -198,8 +198,8 @@ module {
%106 = affine.load %68[%arg1, %103, %104, %105] : memref<1x32x34x34xf32>
%107 = affine.load %73[%102, %arg6, %arg7, %arg8] : memref<32x1x3x3xf32>
%108 = affine.load %67[%arg1, %102, %arg4, %arg5] : memref<1x32x32x32xf32>
%109 = mulf %106, %107 : f32
%110 = addf %108, %109 : f32
%109 = arith.mulf %106, %107 : f32
%110 = arith.addf %108, %109 : f32
affine.store %110, %67[%arg1, %102, %arg4, %arg5] : memref<1x32x32x32xf32>
}
}
@ -214,14 +214,14 @@ module {
affine.for %arg3 = 0 to 32 {
affine.for %arg4 = 0 to 32 {
%102 = affine.load %67[%arg1, %arg2, %arg3, %arg4] : memref<1x32x32x32xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %66[%arg1, %arg2, %arg3, %arg4] : memref<1x32x32x32xf32>
}
}
}
}
%74 = "krnl.global"() {name = "constant_2", offset = 4608 : i64, shape = [64, 32, 1, 1]} : () -> memref<64x32x1x1xf32>
%74 = "krnl.global"() {name = "arith.constant_2", offset = 4608 : i64, shape = [64, 32, 1, 1]} : () -> memref<64x32x1x1xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 64 {
affine.for %arg3 = 0 to 32 {
@ -235,8 +235,8 @@ module {
%104 = affine.load %66[%arg1, %arg5, %102, %103] : memref<1x32x32x32xf32>
%105 = affine.load %74[%arg2, %arg5, %arg6, %arg7] : memref<64x32x1x1xf32>
%106 = affine.load %65[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
%107 = mulf %104, %105 : f32
%108 = addf %106, %107 : f32
%107 = arith.mulf %104, %105 : f32
%108 = arith.addf %106, %107 : f32
affine.store %108, %65[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
}
}
@ -250,14 +250,14 @@ module {
affine.for %arg3 = 0 to 32 {
affine.for %arg4 = 0 to 32 {
%102 = affine.load %65[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %64[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
}
}
}
}
%75 = "krnl.global"() {name = "constant_3", offset = 12800 : i64, shape = [64, 1, 3, 3]} : () -> memref<64x1x3x3xf32>
%75 = "krnl.global"() {name = "arith.constant_3", offset = 12800 : i64, shape = [64, 1, 3, 3]} : () -> memref<64x1x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 64 {
affine.for %arg3 = 0 to 34 {
@ -295,8 +295,8 @@ module {
%106 = affine.load %63[%arg1, %103, %104, %105] : memref<1x64x34x34xf32>
%107 = affine.load %75[%102, %arg6, %arg7, %arg8] : memref<64x1x3x3xf32>
%108 = affine.load %62[%arg1, %102, %arg4, %arg5] : memref<1x64x16x16xf32>
%109 = mulf %106, %107 : f32
%110 = addf %108, %109 : f32
%109 = arith.mulf %106, %107 : f32
%110 = arith.addf %108, %109 : f32
affine.store %110, %62[%arg1, %102, %arg4, %arg5] : memref<1x64x16x16xf32>
}
}
@ -311,14 +311,14 @@ module {
affine.for %arg3 = 0 to 16 {
affine.for %arg4 = 0 to 16 {
%102 = affine.load %62[%arg1, %arg2, %arg3, %arg4] : memref<1x64x16x16xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %61[%arg1, %arg2, %arg3, %arg4] : memref<1x64x16x16xf32>
}
}
}
}
%76 = "krnl.global"() {name = "constant_4", offset = 15104 : i64, shape = [128, 64, 1, 1]} : () -> memref<128x64x1x1xf32>
%76 = "krnl.global"() {name = "arith.constant_4", offset = 15104 : i64, shape = [128, 64, 1, 1]} : () -> memref<128x64x1x1xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 128 {
affine.for %arg3 = 0 to 16 {
@ -332,8 +332,8 @@ module {
%104 = affine.load %61[%arg1, %arg5, %102, %103] : memref<1x64x16x16xf32>
%105 = affine.load %76[%arg2, %arg5, %arg6, %arg7] : memref<128x64x1x1xf32>
%106 = affine.load %60[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
%107 = mulf %104, %105 : f32
%108 = addf %106, %107 : f32
%107 = arith.mulf %104, %105 : f32
%108 = arith.addf %106, %107 : f32
affine.store %108, %60[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
}
}
@ -347,14 +347,14 @@ module {
affine.for %arg3 = 0 to 16 {
affine.for %arg4 = 0 to 16 {
%102 = affine.load %60[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %59[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
}
}
}
}
%77 = "krnl.global"() {name = "constant_5", offset = 47872 : i64, shape = [128, 1, 3, 3]} : () -> memref<128x1x3x3xf32>
%77 = "krnl.global"() {name = "arith.constant_5", offset = 47872 : i64, shape = [128, 1, 3, 3]} : () -> memref<128x1x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 128 {
affine.for %arg3 = 0 to 18 {
@ -392,8 +392,8 @@ module {
%106 = affine.load %58[%arg1, %103, %104, %105] : memref<1x128x18x18xf32>
%107 = affine.load %77[%102, %arg6, %arg7, %arg8] : memref<128x1x3x3xf32>
%108 = affine.load %57[%arg1, %102, %arg4, %arg5] : memref<1x128x16x16xf32>
%109 = mulf %106, %107 : f32
%110 = addf %108, %109 : f32
%109 = arith.mulf %106, %107 : f32
%110 = arith.addf %108, %109 : f32
affine.store %110, %57[%arg1, %102, %arg4, %arg5] : memref<1x128x16x16xf32>
}
}
@ -408,14 +408,14 @@ module {
affine.for %arg3 = 0 to 16 {
affine.for %arg4 = 0 to 16 {
%102 = affine.load %57[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %56[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
}
}
}
}
%78 = "krnl.global"() {name = "constant_6", offset = 52480 : i64, shape = [128, 128, 1, 1]} : () -> memref<128x128x1x1xf32>
%78 = "krnl.global"() {name = "arith.constant_6", offset = 52480 : i64, shape = [128, 128, 1, 1]} : () -> memref<128x128x1x1xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 128 {
affine.for %arg3 = 0 to 16 {
@ -429,8 +429,8 @@ module {
%104 = affine.load %56[%arg1, %arg5, %102, %103] : memref<1x128x16x16xf32>
%105 = affine.load %78[%arg2, %arg5, %arg6, %arg7] : memref<128x128x1x1xf32>
%106 = affine.load %55[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
%107 = mulf %104, %105 : f32
%108 = addf %106, %107 : f32
%107 = arith.mulf %104, %105 : f32
%108 = arith.addf %106, %107 : f32
affine.store %108, %55[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
}
}
@ -444,14 +444,14 @@ module {
affine.for %arg3 = 0 to 16 {
affine.for %arg4 = 0 to 16 {
%102 = affine.load %55[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %54[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
}
}
}
}
%79 = "krnl.global"() {name = "constant_7", offset = 118016 : i64, shape = [128, 1, 3, 3]} : () -> memref<128x1x3x3xf32>
%79 = "krnl.global"() {name = "arith.constant_7", offset = 118016 : i64, shape = [128, 1, 3, 3]} : () -> memref<128x1x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 128 {
affine.for %arg3 = 0 to 18 {
@ -489,8 +489,8 @@ module {
%106 = affine.load %53[%arg1, %103, %104, %105] : memref<1x128x18x18xf32>
%107 = affine.load %79[%102, %arg6, %arg7, %arg8] : memref<128x1x3x3xf32>
%108 = affine.load %52[%arg1, %102, %arg4, %arg5] : memref<1x128x8x8xf32>
%109 = mulf %106, %107 : f32
%110 = addf %108, %109 : f32
%109 = arith.mulf %106, %107 : f32
%110 = arith.addf %108, %109 : f32
affine.store %110, %52[%arg1, %102, %arg4, %arg5] : memref<1x128x8x8xf32>
}
}
@ -505,14 +505,14 @@ module {
affine.for %arg3 = 0 to 8 {
affine.for %arg4 = 0 to 8 {
%102 = affine.load %52[%arg1, %arg2, %arg3, %arg4] : memref<1x128x8x8xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %51[%arg1, %arg2, %arg3, %arg4] : memref<1x128x8x8xf32>
}
}
}
}
%80 = "krnl.global"() {name = "constant_8", offset = 122624 : i64, shape = [256, 128, 1, 1]} : () -> memref<256x128x1x1xf32>
%80 = "krnl.global"() {name = "arith.constant_8", offset = 122624 : i64, shape = [256, 128, 1, 1]} : () -> memref<256x128x1x1xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 256 {
affine.for %arg3 = 0 to 8 {
@ -526,8 +526,8 @@ module {
%104 = affine.load %51[%arg1, %arg5, %102, %103] : memref<1x128x8x8xf32>
%105 = affine.load %80[%arg2, %arg5, %arg6, %arg7] : memref<256x128x1x1xf32>
%106 = affine.load %50[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
%107 = mulf %104, %105 : f32
%108 = addf %106, %107 : f32
%107 = arith.mulf %104, %105 : f32
%108 = arith.addf %106, %107 : f32
affine.store %108, %50[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
}
}
@ -541,14 +541,14 @@ module {
affine.for %arg3 = 0 to 8 {
affine.for %arg4 = 0 to 8 {
%102 = affine.load %50[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %49[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
}
}
}
}
%81 = "krnl.global"() {name = "constant_9", offset = 253696 : i64, shape = [256, 1, 3, 3]} : () -> memref<256x1x3x3xf32>
%81 = "krnl.global"() {name = "arith.constant_9", offset = 253696 : i64, shape = [256, 1, 3, 3]} : () -> memref<256x1x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 256 {
affine.for %arg3 = 0 to 10 {
@ -586,8 +586,8 @@ module {
%106 = affine.load %48[%arg1, %103, %104, %105] : memref<1x256x10x10xf32>
%107 = affine.load %81[%102, %arg6, %arg7, %arg8] : memref<256x1x3x3xf32>
%108 = affine.load %47[%arg1, %102, %arg4, %arg5] : memref<1x256x8x8xf32>
%109 = mulf %106, %107 : f32
%110 = addf %108, %109 : f32
%109 = arith.mulf %106, %107 : f32
%110 = arith.addf %108, %109 : f32
affine.store %110, %47[%arg1, %102, %arg4, %arg5] : memref<1x256x8x8xf32>
}
}
@ -602,14 +602,14 @@ module {
affine.for %arg3 = 0 to 8 {
affine.for %arg4 = 0 to 8 {
%102 = affine.load %47[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %46[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
}
}
}
}
%82 = "krnl.global"() {name = "constant_10", offset = 262912 : i64, shape = [256, 256, 1, 1]} : () -> memref<256x256x1x1xf32>
%82 = "krnl.global"() {name = "arith.constant_10", offset = 262912 : i64, shape = [256, 256, 1, 1]} : () -> memref<256x256x1x1xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 256 {
affine.for %arg3 = 0 to 8 {
@ -623,8 +623,8 @@ module {
%104 = affine.load %46[%arg1, %arg5, %102, %103] : memref<1x256x8x8xf32>
%105 = affine.load %82[%arg2, %arg5, %arg6, %arg7] : memref<256x256x1x1xf32>
%106 = affine.load %45[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
%107 = mulf %104, %105 : f32
%108 = addf %106, %107 : f32
%107 = arith.mulf %104, %105 : f32
%108 = arith.addf %106, %107 : f32
affine.store %108, %45[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
}
}
@ -638,14 +638,14 @@ module {
affine.for %arg3 = 0 to 8 {
affine.for %arg4 = 0 to 8 {
%102 = affine.load %45[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %44[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
}
}
}
}
%83 = "krnl.global"() {name = "constant_11", offset = 525056 : i64, shape = [256, 1, 3, 3]} : () -> memref<256x1x3x3xf32>
%83 = "krnl.global"() {name = "arith.constant_11", offset = 525056 : i64, shape = [256, 1, 3, 3]} : () -> memref<256x1x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 256 {
affine.for %arg3 = 0 to 10 {
@ -683,8 +683,8 @@ module {
%106 = affine.load %43[%arg1, %103, %104, %105] : memref<1x256x10x10xf32>
%107 = affine.load %83[%102, %arg6, %arg7, %arg8] : memref<256x1x3x3xf32>
%108 = affine.load %42[%arg1, %102, %arg4, %arg5] : memref<1x256x4x4xf32>
%109 = mulf %106, %107 : f32
%110 = addf %108, %109 : f32
%109 = arith.mulf %106, %107 : f32
%110 = arith.addf %108, %109 : f32
affine.store %110, %42[%arg1, %102, %arg4, %arg5] : memref<1x256x4x4xf32>
}
}
@ -699,14 +699,14 @@ module {
affine.for %arg3 = 0 to 4 {
affine.for %arg4 = 0 to 4 {
%102 = affine.load %42[%arg1, %arg2, %arg3, %arg4] : memref<1x256x4x4xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %41[%arg1, %arg2, %arg3, %arg4] : memref<1x256x4x4xf32>
}
}
}
}
%84 = "krnl.global"() {name = "constant_12", offset = 534272 : i64, shape = [512, 256, 1, 1]} : () -> memref<512x256x1x1xf32>
%84 = "krnl.global"() {name = "arith.constant_12", offset = 534272 : i64, shape = [512, 256, 1, 1]} : () -> memref<512x256x1x1xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 512 {
affine.for %arg3 = 0 to 4 {
@ -720,8 +720,8 @@ module {
%104 = affine.load %41[%arg1, %arg5, %102, %103] : memref<1x256x4x4xf32>
%105 = affine.load %84[%arg2, %arg5, %arg6, %arg7] : memref<512x256x1x1xf32>
%106 = affine.load %40[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%107 = mulf %104, %105 : f32
%108 = addf %106, %107 : f32
%107 = arith.mulf %104, %105 : f32
%108 = arith.addf %106, %107 : f32
affine.store %108, %40[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
@ -735,14 +735,14 @@ module {
affine.for %arg3 = 0 to 4 {
affine.for %arg4 = 0 to 4 {
%102 = affine.load %40[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %39[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
}
}
%85 = "krnl.global"() {name = "constant_13", offset = 1058560 : i64, shape = [512, 1, 3, 3]} : () -> memref<512x1x3x3xf32>
%85 = "krnl.global"() {name = "arith.constant_13", offset = 1058560 : i64, shape = [512, 1, 3, 3]} : () -> memref<512x1x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 512 {
affine.for %arg3 = 0 to 6 {
@ -780,8 +780,8 @@ module {
%106 = affine.load %38[%arg1, %103, %104, %105] : memref<1x512x6x6xf32>
%107 = affine.load %85[%102, %arg6, %arg7, %arg8] : memref<512x1x3x3xf32>
%108 = affine.load %37[%arg1, %102, %arg4, %arg5] : memref<1x512x4x4xf32>
%109 = mulf %106, %107 : f32
%110 = addf %108, %109 : f32
%109 = arith.mulf %106, %107 : f32
%110 = arith.addf %108, %109 : f32
affine.store %110, %37[%arg1, %102, %arg4, %arg5] : memref<1x512x4x4xf32>
}
}
@ -796,14 +796,14 @@ module {
affine.for %arg3 = 0 to 4 {
affine.for %arg4 = 0 to 4 {
%102 = affine.load %37[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %36[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
}
}
%86 = "krnl.global"() {name = "constant_14", offset = 1076992 : i64, shape = [512, 512, 1, 1]} : () -> memref<512x512x1x1xf32>
%86 = "krnl.global"() {name = "arith.constant_14", offset = 1076992 : i64, shape = [512, 512, 1, 1]} : () -> memref<512x512x1x1xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 512 {
affine.for %arg3 = 0 to 4 {
@ -817,8 +817,8 @@ module {
%104 = affine.load %36[%arg1, %arg5, %102, %103] : memref<1x512x4x4xf32>
%105 = affine.load %86[%arg2, %arg5, %arg6, %arg7] : memref<512x512x1x1xf32>
%106 = affine.load %35[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%107 = mulf %104, %105 : f32
%108 = addf %106, %107 : f32
%107 = arith.mulf %104, %105 : f32
%108 = arith.addf %106, %107 : f32
affine.store %108, %35[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
@ -832,14 +832,14 @@ module {
affine.for %arg3 = 0 to 4 {
affine.for %arg4 = 0 to 4 {
%102 = affine.load %35[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %34[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
}
}
%87 = "krnl.global"() {name = "constant_15", offset = 2125568 : i64, shape = [512, 1, 3, 3]} : () -> memref<512x1x3x3xf32>
%87 = "krnl.global"() {name = "arith.constant_15", offset = 2125568 : i64, shape = [512, 1, 3, 3]} : () -> memref<512x1x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 512 {
affine.for %arg3 = 0 to 6 {
@ -877,8 +877,8 @@ module {
%106 = affine.load %33[%arg1, %103, %104, %105] : memref<1x512x6x6xf32>
%107 = affine.load %87[%102, %arg6, %arg7, %arg8] : memref<512x1x3x3xf32>
%108 = affine.load %32[%arg1, %102, %arg4, %arg5] : memref<1x512x4x4xf32>
%109 = mulf %106, %107 : f32
%110 = addf %108, %109 : f32
%109 = arith.mulf %106, %107 : f32
%110 = arith.addf %108, %109 : f32
affine.store %110, %32[%arg1, %102, %arg4, %arg5] : memref<1x512x4x4xf32>
}
}
@ -893,14 +893,14 @@ module {
affine.for %arg3 = 0 to 4 {
affine.for %arg4 = 0 to 4 {
%102 = affine.load %32[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %31[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
}
}
%88 = "krnl.global"() {name = "constant_16", offset = 2144000 : i64, shape = [512, 512, 1, 1]} : () -> memref<512x512x1x1xf32>
%88 = "krnl.global"() {name = "arith.constant_16", offset = 2144000 : i64, shape = [512, 512, 1, 1]} : () -> memref<512x512x1x1xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 512 {
affine.for %arg3 = 0 to 4 {
@ -914,8 +914,8 @@ module {
%104 = affine.load %31[%arg1, %arg5, %102, %103] : memref<1x512x4x4xf32>
%105 = affine.load %88[%arg2, %arg5, %arg6, %arg7] : memref<512x512x1x1xf32>
%106 = affine.load %30[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%107 = mulf %104, %105 : f32
%108 = addf %106, %107 : f32
%107 = arith.mulf %104, %105 : f32
%108 = arith.addf %106, %107 : f32
affine.store %108, %30[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
@ -929,14 +929,14 @@ module {
affine.for %arg3 = 0 to 4 {
affine.for %arg4 = 0 to 4 {
%102 = affine.load %30[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %29[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
}
}
%89 = "krnl.global"() {name = "constant_17", offset = 3192576 : i64, shape = [512, 1, 3, 3]} : () -> memref<512x1x3x3xf32>
%89 = "krnl.global"() {name = "arith.constant_17", offset = 3192576 : i64, shape = [512, 1, 3, 3]} : () -> memref<512x1x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 512 {
affine.for %arg3 = 0 to 6 {
@ -974,8 +974,8 @@ module {
%106 = affine.load %28[%arg1, %103, %104, %105] : memref<1x512x6x6xf32>
%107 = affine.load %89[%102, %arg6, %arg7, %arg8] : memref<512x1x3x3xf32>
%108 = affine.load %27[%arg1, %102, %arg4, %arg5] : memref<1x512x4x4xf32>
%109 = mulf %106, %107 : f32
%110 = addf %108, %109 : f32
%109 = arith.mulf %106, %107 : f32
%110 = arith.addf %108, %109 : f32
affine.store %110, %27[%arg1, %102, %arg4, %arg5] : memref<1x512x4x4xf32>
}
}
@ -990,14 +990,14 @@ module {
affine.for %arg3 = 0 to 4 {
affine.for %arg4 = 0 to 4 {
%102 = affine.load %27[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %26[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
}
}
%90 = "krnl.global"() {name = "constant_18", offset = 3211008 : i64, shape = [512, 512, 1, 1]} : () -> memref<512x512x1x1xf32>
%90 = "krnl.global"() {name = "arith.constant_18", offset = 3211008 : i64, shape = [512, 512, 1, 1]} : () -> memref<512x512x1x1xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 512 {
affine.for %arg3 = 0 to 4 {
@ -1011,8 +1011,8 @@ module {
%104 = affine.load %26[%arg1, %arg5, %102, %103] : memref<1x512x4x4xf32>
%105 = affine.load %90[%arg2, %arg5, %arg6, %arg7] : memref<512x512x1x1xf32>
%106 = affine.load %25[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%107 = mulf %104, %105 : f32
%108 = addf %106, %107 : f32
%107 = arith.mulf %104, %105 : f32
%108 = arith.addf %106, %107 : f32
affine.store %108, %25[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
@ -1026,14 +1026,14 @@ module {
affine.for %arg3 = 0 to 4 {
affine.for %arg4 = 0 to 4 {
%102 = affine.load %25[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %24[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
}
}
%91 = "krnl.global"() {name = "constant_19", offset = 4259584 : i64, shape = [512, 1, 3, 3]} : () -> memref<512x1x3x3xf32>
%91 = "krnl.global"() {name = "arith.constant_19", offset = 4259584 : i64, shape = [512, 1, 3, 3]} : () -> memref<512x1x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 512 {
affine.for %arg3 = 0 to 6 {
@ -1071,8 +1071,8 @@ module {
%106 = affine.load %23[%arg1, %103, %104, %105] : memref<1x512x6x6xf32>
%107 = affine.load %91[%102, %arg6, %arg7, %arg8] : memref<512x1x3x3xf32>
%108 = affine.load %22[%arg1, %102, %arg4, %arg5] : memref<1x512x4x4xf32>
%109 = mulf %106, %107 : f32
%110 = addf %108, %109 : f32
%109 = arith.mulf %106, %107 : f32
%110 = arith.addf %108, %109 : f32
affine.store %110, %22[%arg1, %102, %arg4, %arg5] : memref<1x512x4x4xf32>
}
}
@ -1087,14 +1087,14 @@ module {
affine.for %arg3 = 0 to 4 {
affine.for %arg4 = 0 to 4 {
%102 = affine.load %22[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %21[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
}
}
%92 = "krnl.global"() {name = "constant_20", offset = 4278016 : i64, shape = [512, 512, 1, 1]} : () -> memref<512x512x1x1xf32>
%92 = "krnl.global"() {name = "arith.constant_20", offset = 4278016 : i64, shape = [512, 512, 1, 1]} : () -> memref<512x512x1x1xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 512 {
affine.for %arg3 = 0 to 4 {
@ -1108,8 +1108,8 @@ module {
%104 = affine.load %21[%arg1, %arg5, %102, %103] : memref<1x512x4x4xf32>
%105 = affine.load %92[%arg2, %arg5, %arg6, %arg7] : memref<512x512x1x1xf32>
%106 = affine.load %20[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%107 = mulf %104, %105 : f32
%108 = addf %106, %107 : f32
%107 = arith.mulf %104, %105 : f32
%108 = arith.addf %106, %107 : f32
affine.store %108, %20[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
@ -1123,14 +1123,14 @@ module {
affine.for %arg3 = 0 to 4 {
affine.for %arg4 = 0 to 4 {
%102 = affine.load %20[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %19[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
}
}
%93 = "krnl.global"() {name = "constant_21", offset = 5326592 : i64, shape = [512, 1, 3, 3]} : () -> memref<512x1x3x3xf32>
%93 = "krnl.global"() {name = "arith.constant_21", offset = 5326592 : i64, shape = [512, 1, 3, 3]} : () -> memref<512x1x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 512 {
affine.for %arg3 = 0 to 6 {
@ -1168,8 +1168,8 @@ module {
%106 = affine.load %18[%arg1, %103, %104, %105] : memref<1x512x6x6xf32>
%107 = affine.load %93[%102, %arg6, %arg7, %arg8] : memref<512x1x3x3xf32>
%108 = affine.load %17[%arg1, %102, %arg4, %arg5] : memref<1x512x4x4xf32>
%109 = mulf %106, %107 : f32
%110 = addf %108, %109 : f32
%109 = arith.mulf %106, %107 : f32
%110 = arith.addf %108, %109 : f32
affine.store %110, %17[%arg1, %102, %arg4, %arg5] : memref<1x512x4x4xf32>
}
}
@ -1184,14 +1184,14 @@ module {
affine.for %arg3 = 0 to 4 {
affine.for %arg4 = 0 to 4 {
%102 = affine.load %17[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %16[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
}
}
%94 = "krnl.global"() {name = "constant_22", offset = 5345024 : i64, shape = [512, 512, 1, 1]} : () -> memref<512x512x1x1xf32>
%94 = "krnl.global"() {name = "arith.constant_22", offset = 5345024 : i64, shape = [512, 512, 1, 1]} : () -> memref<512x512x1x1xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 512 {
affine.for %arg3 = 0 to 4 {
@ -1205,8 +1205,8 @@ module {
%104 = affine.load %16[%arg1, %arg5, %102, %103] : memref<1x512x4x4xf32>
%105 = affine.load %94[%arg2, %arg5, %arg6, %arg7] : memref<512x512x1x1xf32>
%106 = affine.load %15[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%107 = mulf %104, %105 : f32
%108 = addf %106, %107 : f32
%107 = arith.mulf %104, %105 : f32
%108 = arith.addf %106, %107 : f32
affine.store %108, %15[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
@ -1220,14 +1220,14 @@ module {
affine.for %arg3 = 0 to 4 {
affine.for %arg4 = 0 to 4 {
%102 = affine.load %15[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %14[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
}
}
%95 = "krnl.global"() {name = "constant_23", offset = 6393600 : i64, shape = [512, 1, 3, 3]} : () -> memref<512x1x3x3xf32>
%95 = "krnl.global"() {name = "arith.constant_23", offset = 6393600 : i64, shape = [512, 1, 3, 3]} : () -> memref<512x1x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 512 {
affine.for %arg3 = 0 to 6 {
@ -1265,8 +1265,8 @@ module {
%106 = affine.load %13[%arg1, %103, %104, %105] : memref<1x512x6x6xf32>
%107 = affine.load %95[%102, %arg6, %arg7, %arg8] : memref<512x1x3x3xf32>
%108 = affine.load %12[%arg1, %102, %arg4, %arg5] : memref<1x512x2x2xf32>
%109 = mulf %106, %107 : f32
%110 = addf %108, %109 : f32
%109 = arith.mulf %106, %107 : f32
%110 = arith.addf %108, %109 : f32
affine.store %110, %12[%arg1, %102, %arg4, %arg5] : memref<1x512x2x2xf32>
}
}
@ -1281,14 +1281,14 @@ module {
affine.for %arg3 = 0 to 2 {
affine.for %arg4 = 0 to 2 {
%102 = affine.load %12[%arg1, %arg2, %arg3, %arg4] : memref<1x512x2x2xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %11[%arg1, %arg2, %arg3, %arg4] : memref<1x512x2x2xf32>
}
}
}
}
%96 = "krnl.global"() {name = "constant_24", offset = 6412032 : i64, shape = [1024, 512, 1, 1]} : () -> memref<1024x512x1x1xf32>
%96 = "krnl.global"() {name = "arith.constant_24", offset = 6412032 : i64, shape = [1024, 512, 1, 1]} : () -> memref<1024x512x1x1xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 1024 {
affine.for %arg3 = 0 to 2 {
@ -1302,8 +1302,8 @@ module {
%104 = affine.load %11[%arg1, %arg5, %102, %103] : memref<1x512x2x2xf32>
%105 = affine.load %96[%arg2, %arg5, %arg6, %arg7] : memref<1024x512x1x1xf32>
%106 = affine.load %10[%arg1, %arg2, %arg3, %arg4] : memref<1x1024x2x2xf32>
%107 = mulf %104, %105 : f32
%108 = addf %106, %107 : f32
%107 = arith.mulf %104, %105 : f32
%108 = arith.addf %106, %107 : f32
affine.store %108, %10[%arg1, %arg2, %arg3, %arg4] : memref<1x1024x2x2xf32>
}
}
@ -1317,14 +1317,14 @@ module {
affine.for %arg3 = 0 to 2 {
affine.for %arg4 = 0 to 2 {
%102 = affine.load %10[%arg1, %arg2, %arg3, %arg4] : memref<1x1024x2x2xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %9[%arg1, %arg2, %arg3, %arg4] : memref<1x1024x2x2xf32>
}
}
}
}
%97 = "krnl.global"() {name = "constant_25", offset = 8509184 : i64, shape = [1024, 1, 3, 3]} : () -> memref<1024x1x3x3xf32>
%97 = "krnl.global"() {name = "arith.constant_25", offset = 8509184 : i64, shape = [1024, 1, 3, 3]} : () -> memref<1024x1x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 1024 {
affine.for %arg3 = 0 to 4 {
@ -1362,8 +1362,8 @@ module {
%106 = affine.load %8[%arg1, %103, %104, %105] : memref<1x1024x4x4xf32>
%107 = affine.load %97[%102, %arg6, %arg7, %arg8] : memref<1024x1x3x3xf32>
%108 = affine.load %7[%arg1, %102, %arg4, %arg5] : memref<1x1024x2x2xf32>
%109 = mulf %106, %107 : f32
%110 = addf %108, %109 : f32
%109 = arith.mulf %106, %107 : f32
%110 = arith.addf %108, %109 : f32
affine.store %110, %7[%arg1, %102, %arg4, %arg5] : memref<1x1024x2x2xf32>
}
}
@ -1378,14 +1378,14 @@ module {
affine.for %arg3 = 0 to 2 {
affine.for %arg4 = 0 to 2 {
%102 = affine.load %7[%arg1, %arg2, %arg3, %arg4] : memref<1x1024x2x2xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %6[%arg1, %arg2, %arg3, %arg4] : memref<1x1024x2x2xf32>
}
}
}
}
%98 = "krnl.global"() {name = "constant_26", offset = 8546048 : i64, shape = [1024, 1024, 1, 1]} : () -> memref<1024x1024x1x1xf32>
%98 = "krnl.global"() {name = "arith.constant_26", offset = 8546048 : i64, shape = [1024, 1024, 1, 1]} : () -> memref<1024x1024x1x1xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 1024 {
affine.for %arg3 = 0 to 2 {
@ -1399,8 +1399,8 @@ module {
%104 = affine.load %6[%arg1, %arg5, %102, %103] : memref<1x1024x2x2xf32>
%105 = affine.load %98[%arg2, %arg5, %arg6, %arg7] : memref<1024x1024x1x1xf32>
%106 = affine.load %5[%arg1, %arg2, %arg3, %arg4] : memref<1x1024x2x2xf32>
%107 = mulf %104, %105 : f32
%108 = addf %106, %107 : f32
%107 = arith.mulf %104, %105 : f32
%108 = arith.addf %106, %107 : f32
affine.store %108, %5[%arg1, %arg2, %arg3, %arg4] : memref<1x1024x2x2xf32>
}
}
@ -1414,7 +1414,7 @@ module {
affine.for %arg3 = 0 to 2 {
affine.for %arg4 = 0 to 2 {
%102 = affine.load %5[%arg1, %arg2, %arg3, %arg4] : memref<1x1024x2x2xf32>
%103 = cmpf "olt", %102, %cst_0 : f32
%103 = arith.cmpf "olt", %102, %cst_0 : f32
%104 = select %103, %cst_0, %102 : f32
affine.store %104, %4[%arg1, %arg2, %arg3, %arg4] : memref<1x1024x2x2xf32>
}
@ -1436,19 +1436,19 @@ module {
affine.for %arg4 = 0 to 2 {
%102 = affine.load %4[%arg1, %arg2, %arg3, %arg4] : memref<1x1024x2x2xf32>
%103 = affine.load %3[%arg1, %arg2, %c0, %c0] : memref<1x1024x1x1xf32>
%104 = addf %103, %102 : f32
%104 = arith.addf %103, %102 : f32
affine.store %104, %3[%arg1, %arg2, %c0, %c0] : memref<1x1024x1x1xf32>
}
}
}
}
%99 = uitofp %c4_i64 : i64 to f32
%99 = arith.uitofp %c4_i64 : i64 to f32
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 1024 {
affine.for %arg3 = 0 to 1 {
affine.for %arg4 = 0 to 1 {
%102 = affine.load %3[%arg1, %arg2, %arg3, %arg4] : memref<1x1024x1x1xf32>
%103 = divf %102, %99 : f32
%103 = arith.divf %102, %99 : f32
affine.store %103, %3[%arg1, %arg2, %arg3, %arg4] : memref<1x1024x1x1xf32>
}
}
@ -1465,8 +1465,8 @@ module {
}
}
}
%100 = "krnl.global"() {name = "constant_27", offset = 12740352 : i64, shape = [10, 1024]} : () -> memref<10x1024xf32>
%101 = "krnl.global"() {name = "constant_28", offset = 12781312 : i64, shape = [10], value = dense<[-0.0178910494, 0.00394501537, 0.0126184784, -0.0173512287, -0.00768596306, -0.0264055394, 0.00475354865, -0.0243351571, -0.0252915286, 0.0226600058]> : tensor<10xf32>} : () -> memref<10xf32>
%100 = "krnl.global"() {name = "arith.constant_27", offset = 12740352 : i64, shape = [10, 1024]} : () -> memref<10x1024xf32>
%101 = "krnl.global"() {name = "arith.constant_28", offset = 12781312 : i64, shape = [10], value = dense<[-0.0178910494, 0.00394501537, 0.0126184784, -0.0173512287, -0.00768596306, -0.0264055394, 0.00475354865, -0.0243351571, -0.0252915286, 0.0226600058]> : tensor<10xf32>} : () -> memref<10xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 10 {
affine.store %cst_0, %1[%arg1, %arg2] : memref<1x10xf32>
@ -1474,15 +1474,15 @@ module {
%107 = affine.load %2[%arg1, %arg3] : memref<1x1024xf32>
%108 = affine.load %100[%arg2, %arg3] : memref<10x1024xf32>
%109 = affine.load %1[%arg1, %arg2] : memref<1x10xf32>
%110 = mulf %107, %108 : f32
%111 = addf %109, %110 : f32
%110 = arith.mulf %107, %108 : f32
%111 = arith.addf %109, %110 : f32
affine.store %111, %1[%arg1, %arg2] : memref<1x10xf32>
}
%102 = affine.load %1[%arg1, %arg2] : memref<1x10xf32>
%103 = mulf %cst, %102 : f32
%103 = arith.mulf %cst, %102 : f32
%104 = affine.load %101[%arg2] : memref<10xf32>
%105 = mulf %cst, %104 : f32
%106 = addf %103, %105 : f32
%105 = arith.mulf %cst, %104 : f32
%106 = arith.addf %103, %105 : f32
affine.store %106, %1[%arg1, %arg2] : memref<1x10xf32>
}
}

474
samples/onnx-mlir/mobilenetv2/mobilenetv2.mlir
View File

File diff suppressed because it is too large Load Diff

202
samples/onnx-mlir/resnet18/resnet18.mlir
View File

@ -23,12 +23,12 @@
module {
%0 = "krnl.packed_const"() {file_name = "/tmp/packed_const-175e48.tmp", is_le = true, size_in_bytes = 44657408 : i64} : () -> i64
func @main_graph(%arg0: memref<1x3x32x32xf32>) -> memref<1x10xf32> attributes {input_names = ["input.1"], output_names = ["70"]} {
%c0 = constant 0 : index
%c16_i64 = constant 16 : i64
%c512 = constant 512 : index
%c1 = constant 1 : index
%cst = constant 1.000000e+00 : f32
%cst_0 = constant 0.000000e+00 : f32
%c0 = arith.constant 0 : index
%c16_i64 = arith.constant 16 : i64
%c512 = arith.constant 512 : index
%c1 = arith.constant 1 : index
%cst = arith.constant 1.000000e+00 : f32
%cst_0 = arith.constant 0.000000e+00 : f32
%1 = memref.alloc() : memref<1x10xf32>
%2 = memref.alloc() : memref<1x512xf32>
%3 = memref.alloc() : memref<1x512x1x1xf32>
@ -94,7 +94,7 @@ module {
%63 = memref.alloc() : memref<1x64x32x32xf32>
%64 = memref.alloc() : memref<1x64x32x32xf32>
%65 = memref.alloc() : memref<1x3x34x34xf32>
%66 = "krnl.global"() {name = "constant_0", offset = 0 : i64, shape = [64, 3, 3, 3]} : () -> memref<64x3x3x3xf32>
%66 = "krnl.global"() {name = "arith.constant_0", offset = 0 : i64, shape = [64, 3, 3, 3]} : () -> memref<64x3x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 3 {
affine.for %arg3 = 0 to 34 {
@ -129,8 +129,8 @@ module {
%91 = affine.load %65[%arg1, %arg5, %89, %90] : memref<1x3x34x34xf32>
%92 = affine.load %66[%arg2, %arg5, %arg6, %arg7] : memref<64x3x3x3xf32>
%93 = affine.load %64[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
%94 = mulf %91, %92 : f32
%95 = addf %93, %94 : f32
%94 = arith.mulf %91, %92 : f32
%95 = arith.addf %93, %94 : f32
affine.store %95, %64[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
}
}
@ -144,14 +144,14 @@ module {
affine.for %arg3 = 0 to 32 {
affine.for %arg4 = 0 to 32 {
%89 = affine.load %64[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
%90 = cmpf "olt", %89, %cst_0 : f32
%90 = arith.cmpf "olt", %89, %cst_0 : f32
%91 = select %90, %cst_0, %89 : f32
affine.store %91, %63[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
}
}
}
}
%67 = "krnl.global"() {name = "constant_1", offset = 6912 : i64, shape = [64, 64, 3, 3]} : () -> memref<64x64x3x3xf32>
%67 = "krnl.global"() {name = "arith.constant_1", offset = 6912 : i64, shape = [64, 64, 3, 3]} : () -> memref<64x64x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 64 {
affine.for %arg3 = 0 to 34 {
@ -186,8 +186,8 @@ module {
%91 = affine.load %62[%arg1, %arg5, %89, %90] : memref<1x64x34x34xf32>
%92 = affine.load %67[%arg2, %arg5, %arg6, %arg7] : memref<64x64x3x3xf32>
%93 = affine.load %61[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
%94 = mulf %91, %92 : f32
%95 = addf %93, %94 : f32
%94 = arith.mulf %91, %92 : f32
%95 = arith.addf %93, %94 : f32
affine.store %95, %61[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
}
}
@ -201,14 +201,14 @@ module {
affine.for %arg3 = 0 to 32 {
affine.for %arg4 = 0 to 32 {
%89 = affine.load %61[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
%90 = cmpf "olt", %89, %cst_0 : f32
%90 = arith.cmpf "olt", %89, %cst_0 : f32
%91 = select %90, %cst_0, %89 : f32
affine.store %91, %60[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
}
}
}
}
%68 = "krnl.global"() {name = "constant_2", offset = 154368 : i64, shape = [64, 64, 3, 3]} : () -> memref<64x64x3x3xf32>
%68 = "krnl.global"() {name = "arith.constant_2", offset = 154368 : i64, shape = [64, 64, 3, 3]} : () -> memref<64x64x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 64 {
affine.for %arg3 = 0 to 34 {
@ -243,8 +243,8 @@ module {
%91 = affine.load %59[%arg1, %arg5, %89, %90] : memref<1x64x34x34xf32>
%92 = affine.load %68[%arg2, %arg5, %arg6, %arg7] : memref<64x64x3x3xf32>
%93 = affine.load %58[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
%94 = mulf %91, %92 : f32
%95 = addf %93, %94 : f32
%94 = arith.mulf %91, %92 : f32
%95 = arith.addf %93, %94 : f32
affine.store %95, %58[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
}
}
@ -259,7 +259,7 @@ module {
affine.for %arg4 = 0 to 32 {
%89 = affine.load %58[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
%90 = affine.load %63[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
%91 = addf %89, %90 : f32
%91 = arith.addf %89, %90 : f32
affine.store %91, %57[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
}
}
@ -270,14 +270,14 @@ module {
affine.for %arg3 = 0 to 32 {
affine.for %arg4 = 0 to 32 {
%89 = affine.load %57[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
%90 = cmpf "olt", %89, %cst_0 : f32
%90 = arith.cmpf "olt", %89, %cst_0 : f32
%91 = select %90, %cst_0, %89 : f32
affine.store %91, %56[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
}
}
}
}
%69 = "krnl.global"() {name = "constant_3", offset = 301824 : i64, shape = [64, 64, 3, 3]} : () -> memref<64x64x3x3xf32>
%69 = "krnl.global"() {name = "arith.constant_3", offset = 301824 : i64, shape = [64, 64, 3, 3]} : () -> memref<64x64x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 64 {
affine.for %arg3 = 0 to 34 {
@ -312,8 +312,8 @@ module {
%91 = affine.load %55[%arg1, %arg5, %89, %90] : memref<1x64x34x34xf32>
%92 = affine.load %69[%arg2, %arg5, %arg6, %arg7] : memref<64x64x3x3xf32>
%93 = affine.load %54[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
%94 = mulf %91, %92 : f32
%95 = addf %93, %94 : f32
%94 = arith.mulf %91, %92 : f32
%95 = arith.addf %93, %94 : f32
affine.store %95, %54[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
}
}
@ -327,14 +327,14 @@ module {
affine.for %arg3 = 0 to 32 {
affine.for %arg4 = 0 to 32 {
%89 = affine.load %54[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
%90 = cmpf "olt", %89, %cst_0 : f32
%90 = arith.cmpf "olt", %89, %cst_0 : f32
%91 = select %90, %cst_0, %89 : f32
affine.store %91, %53[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
}
}
}
}
%70 = "krnl.global"() {name = "constant_4", offset = 449280 : i64, shape = [64, 64, 3, 3]} : () -> memref<64x64x3x3xf32>
%70 = "krnl.global"() {name = "arith.constant_4", offset = 449280 : i64, shape = [64, 64, 3, 3]} : () -> memref<64x64x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 64 {
affine.for %arg3 = 0 to 34 {
@ -369,8 +369,8 @@ module {
%91 = affine.load %52[%arg1, %arg5, %89, %90] : memref<1x64x34x34xf32>
%92 = affine.load %70[%arg2, %arg5, %arg6, %arg7] : memref<64x64x3x3xf32>
%93 = affine.load %51[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
%94 = mulf %91, %92 : f32
%95 = addf %93, %94 : f32
%94 = arith.mulf %91, %92 : f32
%95 = arith.addf %93, %94 : f32
affine.store %95, %51[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
}
}
@ -385,7 +385,7 @@ module {
affine.for %arg4 = 0 to 32 {
%89 = affine.load %51[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
%90 = affine.load %56[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
%91 = addf %89, %90 : f32
%91 = arith.addf %89, %90 : f32
affine.store %91, %50[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
}
}
@ -396,14 +396,14 @@ module {
affine.for %arg3 = 0 to 32 {
affine.for %arg4 = 0 to 32 {
%89 = affine.load %50[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
%90 = cmpf "olt", %89, %cst_0 : f32
%90 = arith.cmpf "olt", %89, %cst_0 : f32
%91 = select %90, %cst_0, %89 : f32
affine.store %91, %49[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
}
}
}
}
%71 = "krnl.global"() {name = "constant_5", offset = 596736 : i64, shape = [128, 64, 3, 3]} : () -> memref<128x64x3x3xf32>
%71 = "krnl.global"() {name = "arith.constant_5", offset = 596736 : i64, shape = [128, 64, 3, 3]} : () -> memref<128x64x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 64 {
affine.for %arg3 = 0 to 34 {
@ -438,8 +438,8 @@ module {
%91 = affine.load %48[%arg1, %arg5, %89, %90] : memref<1x64x34x34xf32>
%92 = affine.load %71[%arg2, %arg5, %arg6, %arg7] : memref<128x64x3x3xf32>
%93 = affine.load %47[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
%94 = mulf %91, %92 : f32
%95 = addf %93, %94 : f32
%94 = arith.mulf %91, %92 : f32
%95 = arith.addf %93, %94 : f32
affine.store %95, %47[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
}
}
@ -453,14 +453,14 @@ module {
affine.for %arg3 = 0 to 16 {
affine.for %arg4 = 0 to 16 {
%89 = affine.load %47[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
%90 = cmpf "olt", %89, %cst_0 : f32
%90 = arith.cmpf "olt", %89, %cst_0 : f32
%91 = select %90, %cst_0, %89 : f32
affine.store %91, %46[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
}
}
}
}
%72 = "krnl.global"() {name = "constant_6", offset = 891648 : i64, shape = [128, 128, 3, 3]} : () -> memref<128x128x3x3xf32>
%72 = "krnl.global"() {name = "arith.constant_6", offset = 891648 : i64, shape = [128, 128, 3, 3]} : () -> memref<128x128x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 128 {
affine.for %arg3 = 0 to 18 {
@ -495,8 +495,8 @@ module {
%91 = affine.load %45[%arg1, %arg5, %89, %90] : memref<1x128x18x18xf32>
%92 = affine.load %72[%arg2, %arg5, %arg6, %arg7] : memref<128x128x3x3xf32>
%93 = affine.load %44[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
%94 = mulf %91, %92 : f32
%95 = addf %93, %94 : f32
%94 = arith.mulf %91, %92 : f32
%95 = arith.addf %93, %94 : f32
affine.store %95, %44[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
}
}
@ -505,7 +505,7 @@ module {
}
}
}
%73 = "krnl.global"() {name = "constant_7", offset = 1481472 : i64, shape = [128, 64, 1, 1]} : () -> memref<128x64x1x1xf32>
%73 = "krnl.global"() {name = "arith.constant_7", offset = 1481472 : i64, shape = [128, 64, 1, 1]} : () -> memref<128x64x1x1xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 128 {
affine.for %arg3 = 0 to 16 {
@ -519,8 +519,8 @@ module {
%91 = affine.load %49[%arg1, %arg5, %89, %90] : memref<1x64x32x32xf32>
%92 = affine.load %73[%arg2, %arg5, %arg6, %arg7] : memref<128x64x1x1xf32>
%93 = affine.load %43[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
%94 = mulf %91, %92 : f32
%95 = addf %93, %94 : f32
%94 = arith.mulf %91, %92 : f32
%95 = arith.addf %93, %94 : f32
affine.store %95, %43[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
}
}
@ -535,7 +535,7 @@ module {
affine.for %arg4 = 0 to 16 {
%89 = affine.load %44[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
%90 = affine.load %43[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
%91 = addf %89, %90 : f32
%91 = arith.addf %89, %90 : f32
affine.store %91, %42[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
}
}
@ -546,14 +546,14 @@ module {
affine.for %arg3 = 0 to 16 {
affine.for %arg4 = 0 to 16 {
%89 = affine.load %42[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
%90 = cmpf "olt", %89, %cst_0 : f32
%90 = arith.cmpf "olt", %89, %cst_0 : f32
%91 = select %90, %cst_0, %89 : f32
affine.store %91, %41[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
}
}
}
}
%74 = "krnl.global"() {name = "constant_8", offset = 1514240 : i64, shape = [128, 128, 3, 3]} : () -> memref<128x128x3x3xf32>
%74 = "krnl.global"() {name = "arith.constant_8", offset = 1514240 : i64, shape = [128, 128, 3, 3]} : () -> memref<128x128x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 128 {
affine.for %arg3 = 0 to 18 {
@ -588,8 +588,8 @@ module {
%91 = affine.load %40[%arg1, %arg5, %89, %90] : memref<1x128x18x18xf32>
%92 = affine.load %74[%arg2, %arg5, %arg6, %arg7] : memref<128x128x3x3xf32>
%93 = affine.load %39[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
%94 = mulf %91, %92 : f32
%95 = addf %93, %94 : f32
%94 = arith.mulf %91, %92 : f32
%95 = arith.addf %93, %94 : f32
affine.store %95, %39[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
}
}
@ -603,14 +603,14 @@ module {
affine.for %arg3 = 0 to 16 {
affine.for %arg4 = 0 to 16 {
%89 = affine.load %39[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
%90 = cmpf "olt", %89, %cst_0 : f32
%90 = arith.cmpf "olt", %89, %cst_0 : f32
%91 = select %90, %cst_0, %89 : f32
affine.store %91, %38[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
}
}
}
}
%75 = "krnl.global"() {name = "constant_9", offset = 2104064 : i64, shape = [128, 128, 3, 3]} : () -> memref<128x128x3x3xf32>
%75 = "krnl.global"() {name = "arith.constant_9", offset = 2104064 : i64, shape = [128, 128, 3, 3]} : () -> memref<128x128x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 128 {
affine.for %arg3 = 0 to 18 {
@ -645,8 +645,8 @@ module {
%91 = affine.load %37[%arg1, %arg5, %89, %90] : memref<1x128x18x18xf32>
%92 = affine.load %75[%arg2, %arg5, %arg6, %arg7] : memref<128x128x3x3xf32>
%93 = affine.load %36[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
%94 = mulf %91, %92 : f32
%95 = addf %93, %94 : f32
%94 = arith.mulf %91, %92 : f32
%95 = arith.addf %93, %94 : f32
affine.store %95, %36[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
}
}
@ -661,7 +661,7 @@ module {
affine.for %arg4 = 0 to 16 {
%89 = affine.load %36[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
%90 = affine.load %41[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
%91 = addf %89, %90 : f32
%91 = arith.addf %89, %90 : f32
affine.store %91, %35[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
}
}
@ -672,14 +672,14 @@ module {
affine.for %arg3 = 0 to 16 {
affine.for %arg4 = 0 to 16 {
%89 = affine.load %35[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
%90 = cmpf "olt", %89, %cst_0 : f32
%90 = arith.cmpf "olt", %89, %cst_0 : f32
%91 = select %90, %cst_0, %89 : f32
affine.store %91, %34[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
}
}
}
}
%76 = "krnl.global"() {name = "constant_10", offset = 2693888 : i64, shape = [256, 128, 3, 3]} : () -> memref<256x128x3x3xf32>
%76 = "krnl.global"() {name = "arith.constant_10", offset = 2693888 : i64, shape = [256, 128, 3, 3]} : () -> memref<256x128x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 128 {
affine.for %arg3 = 0 to 18 {
@ -714,8 +714,8 @@ module {
%91 = affine.load %33[%arg1, %arg5, %89, %90] : memref<1x128x18x18xf32>
%92 = affine.load %76[%arg2, %arg5, %arg6, %arg7] : memref<256x128x3x3xf32>
%93 = affine.load %32[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
%94 = mulf %91, %92 : f32
%95 = addf %93, %94 : f32
%94 = arith.mulf %91, %92 : f32
%95 = arith.addf %93, %94 : f32
affine.store %95, %32[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
}
}
@ -729,14 +729,14 @@ module {
affine.for %arg3 = 0 to 8 {
affine.for %arg4 = 0 to 8 {
%89 = affine.load %32[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
%90 = cmpf "olt", %89, %cst_0 : f32
%90 = arith.cmpf "olt", %89, %cst_0 : f32
%91 = select %90, %cst_0, %89 : f32
affine.store %91, %31[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
}
}
}
}
%77 = "krnl.global"() {name = "constant_11", offset = 3873536 : i64, shape = [256, 256, 3, 3]} : () -> memref<256x256x3x3xf32>
%77 = "krnl.global"() {name = "arith.constant_11", offset = 3873536 : i64, shape = [256, 256, 3, 3]} : () -> memref<256x256x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 256 {
affine.for %arg3 = 0 to 10 {
@ -771,8 +771,8 @@ module {
%91 = affine.load %30[%arg1, %arg5, %89, %90] : memref<1x256x10x10xf32>
%92 = affine.load %77[%arg2, %arg5, %arg6, %arg7] : memref<256x256x3x3xf32>
%93 = affine.load %29[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
%94 = mulf %91, %92 : f32
%95 = addf %93, %94 : f32
%94 = arith.mulf %91, %92 : f32
%95 = arith.addf %93, %94 : f32
affine.store %95, %29[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
}
}
@ -781,7 +781,7 @@ module {
}
}
}
%78 = "krnl.global"() {name = "constant_12", offset = 6232832 : i64, shape = [256, 128, 1, 1]} : () -> memref<256x128x1x1xf32>
%78 = "krnl.global"() {name = "arith.constant_12", offset = 6232832 : i64, shape = [256, 128, 1, 1]} : () -> memref<256x128x1x1xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 256 {
affine.for %arg3 = 0 to 8 {
@ -795,8 +795,8 @@ module {
%91 = affine.load %34[%arg1, %arg5, %89, %90] : memref<1x128x16x16xf32>
%92 = affine.load %78[%arg2, %arg5, %arg6, %arg7] : memref<256x128x1x1xf32>
%93 = affine.load %28[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
%94 = mulf %91, %92 : f32
%95 = addf %93, %94 : f32
%94 = arith.mulf %91, %92 : f32
%95 = arith.addf %93, %94 : f32
affine.store %95, %28[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
}
}
@ -811,7 +811,7 @@ module {
affine.for %arg4 = 0 to 8 {
%89 = affine.load %29[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
%90 = affine.load %28[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
%91 = addf %89, %90 : f32
%91 = arith.addf %89, %90 : f32
affine.store %91, %27[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
}
}
@ -822,14 +822,14 @@ module {
affine.for %arg3 = 0 to 8 {
affine.for %arg4 = 0 to 8 {
%89 = affine.load %27[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
%90 = cmpf "olt", %89, %cst_0 : f32
%90 = arith.cmpf "olt", %89, %cst_0 : f32
%91 = select %90, %cst_0, %89 : f32
affine.store %91, %26[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
}
}
}
}
%79 = "krnl.global"() {name = "constant_13", offset = 6363904 : i64, shape = [256, 256, 3, 3]} : () -> memref<256x256x3x3xf32>
%79 = "krnl.global"() {name = "arith.constant_13", offset = 6363904 : i64, shape = [256, 256, 3, 3]} : () -> memref<256x256x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 256 {
affine.for %arg3 = 0 to 10 {
@ -864,8 +864,8 @@ module {
%91 = affine.load %25[%arg1, %arg5, %89, %90] : memref<1x256x10x10xf32>
%92 = affine.load %79[%arg2, %arg5, %arg6, %arg7] : memref<256x256x3x3xf32>
%93 = affine.load %24[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
%94 = mulf %91, %92 : f32
%95 = addf %93, %94 : f32
%94 = arith.mulf %91, %92 : f32
%95 = arith.addf %93, %94 : f32
affine.store %95, %24[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
}
}
@ -879,14 +879,14 @@ module {
affine.for %arg3 = 0 to 8 {
affine.for %arg4 = 0 to 8 {
%89 = affine.load %24[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
%90 = cmpf "olt", %89, %cst_0 : f32
%90 = arith.cmpf "olt", %89, %cst_0 : f32
%91 = select %90, %cst_0, %89 : f32
affine.store %91, %23[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
}
}
}
}
%80 = "krnl.global"() {name = "constant_14", offset = 8723200 : i64, shape = [256, 256, 3, 3]} : () -> memref<256x256x3x3xf32>
%80 = "krnl.global"() {name = "arith.constant_14", offset = 8723200 : i64, shape = [256, 256, 3, 3]} : () -> memref<256x256x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 256 {
affine.for %arg3 = 0 to 10 {
@ -921,8 +921,8 @@ module {
%91 = affine.load %22[%arg1, %arg5, %89, %90] : memref<1x256x10x10xf32>
%92 = affine.load %80[%arg2, %arg5, %arg6, %arg7] : memref<256x256x3x3xf32>
%93 = affine.load %21[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
%94 = mulf %91, %92 : f32
%95 = addf %93, %94 : f32
%94 = arith.mulf %91, %92 : f32
%95 = arith.addf %93, %94 : f32
affine.store %95, %21[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
}
}
@ -937,7 +937,7 @@ module {
affine.for %arg4 = 0 to 8 {
%89 = affine.load %21[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
%90 = affine.load %26[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
%91 = addf %89, %90 : f32
%91 = arith.addf %89, %90 : f32
affine.store %91, %20[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
}
}
@ -948,14 +948,14 @@ module {
affine.for %arg3 = 0 to 8 {
affine.for %arg4 = 0 to 8 {
%89 = affine.load %20[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
%90 = cmpf "olt", %89, %cst_0 : f32
%90 = arith.cmpf "olt", %89, %cst_0 : f32
%91 = select %90, %cst_0, %89 : f32
affine.store %91, %19[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
}
}
}
}
%81 = "krnl.global"() {name = "constant_15", offset = 11082496 : i64, shape = [512, 256, 3, 3]} : () -> memref<512x256x3x3xf32>
%81 = "krnl.global"() {name = "arith.constant_15", offset = 11082496 : i64, shape = [512, 256, 3, 3]} : () -> memref<512x256x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 256 {
affine.for %arg3 = 0 to 10 {
@ -990,8 +990,8 @@ module {
%91 = affine.load %18[%arg1, %arg5, %89, %90] : memref<1x256x10x10xf32>
%92 = affine.load %81[%arg2, %arg5, %arg6, %arg7] : memref<512x256x3x3xf32>
%93 = affine.load %17[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%94 = mulf %91, %92 : f32
%95 = addf %93, %94 : f32
%94 = arith.mulf %91, %92 : f32
%95 = arith.addf %93, %94 : f32
affine.store %95, %17[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
@ -1005,14 +1005,14 @@ module {
affine.for %arg3 = 0 to 4 {
affine.for %arg4 = 0 to 4 {
%89 = affine.load %17[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%90 = cmpf "olt", %89, %cst_0 : f32
%90 = arith.cmpf "olt", %89, %cst_0 : f32
%91 = select %90, %cst_0, %89 : f32
affine.store %91, %16[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
}
}
%82 = "krnl.global"() {name = "constant_16", offset = 15801088 : i64, shape = [512, 512, 3, 3]} : () -> memref<512x512x3x3xf32>
%82 = "krnl.global"() {name = "arith.constant_16", offset = 15801088 : i64, shape = [512, 512, 3, 3]} : () -> memref<512x512x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 512 {
affine.for %arg3 = 0 to 6 {
@ -1047,8 +1047,8 @@ module {
%91 = affine.load %15[%arg1, %arg5, %89, %90] : memref<1x512x6x6xf32>
%92 = affine.load %82[%arg2, %arg5, %arg6, %arg7] : memref<512x512x3x3xf32>
%93 = affine.load %14[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%94 = mulf %91, %92 : f32
%95 = addf %93, %94 : f32
%94 = arith.mulf %91, %92 : f32
%95 = arith.addf %93, %94 : f32
affine.store %95, %14[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
@ -1057,7 +1057,7 @@ module {
}
}
}
%83 = "krnl.global"() {name = "constant_17", offset = 25238272 : i64, shape = [512, 256, 1, 1]} : () -> memref<512x256x1x1xf32>
%83 = "krnl.global"() {name = "arith.constant_17", offset = 25238272 : i64, shape = [512, 256, 1, 1]} : () -> memref<512x256x1x1xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 512 {
affine.for %arg3 = 0 to 4 {
@ -1071,8 +1071,8 @@ module {
%91 = affine.load %19[%arg1, %arg5, %89, %90] : memref<1x256x8x8xf32>
%92 = affine.load %83[%arg2, %arg5, %arg6, %arg7] : memref<512x256x1x1xf32>
%93 = affine.load %13[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%94 = mulf %91, %92 : f32
%95 = addf %93, %94 : f32
%94 = arith.mulf %91, %92 : f32
%95 = arith.addf %93, %94 : f32
affine.store %95, %13[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
@ -1087,7 +1087,7 @@ module {
affine.for %arg4 = 0 to 4 {
%89 = affine.load %14[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%90 = affine.load %13[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%91 = addf %89, %90 : f32
%91 = arith.addf %89, %90 : f32
affine.store %91, %12[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
@ -1098,14 +1098,14 @@ module {
affine.for %arg3 = 0 to 4 {
affine.for %arg4 = 0 to 4 {
%89 = affine.load %12[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%90 = cmpf "olt", %89, %cst_0 : f32
%90 = arith.cmpf "olt", %89, %cst_0 : f32
%91 = select %90, %cst_0, %89 : f32
affine.store %91, %11[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
}
}
%84 = "krnl.global"() {name = "constant_18", offset = 25762560 : i64, shape = [512, 512, 3, 3]} : () -> memref<512x512x3x3xf32>
%84 = "krnl.global"() {name = "arith.constant_18", offset = 25762560 : i64, shape = [512, 512, 3, 3]} : () -> memref<512x512x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 512 {
affine.for %arg3 = 0 to 6 {
@ -1140,8 +1140,8 @@ module {
%91 = affine.load %10[%arg1, %arg5, %89, %90] : memref<1x512x6x6xf32>
%92 = affine.load %84[%arg2, %arg5, %arg6, %arg7] : memref<512x512x3x3xf32>
%93 = affine.load %9[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%94 = mulf %91, %92 : f32
%95 = addf %93, %94 : f32
%94 = arith.mulf %91, %92 : f32
%95 = arith.addf %93, %94 : f32
affine.store %95, %9[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
@ -1155,14 +1155,14 @@ module {
affine.for %arg3 = 0 to 4 {
affine.for %arg4 = 0 to 4 {
%89 = affine.load %9[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%90 = cmpf "olt", %89, %cst_0 : f32
%90 = arith.cmpf "olt", %89, %cst_0 : f32
%91 = select %90, %cst_0, %89 : f32
affine.store %91, %8[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
}
}
%85 = "krnl.global"() {name = "constant_19", offset = 35199744 : i64, shape = [512, 512, 3, 3]} : () -> memref<512x512x3x3xf32>
%85 = "krnl.global"() {name = "arith.constant_19", offset = 35199744 : i64, shape = [512, 512, 3, 3]} : () -> memref<512x512x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 512 {
affine.for %arg3 = 0 to 6 {
@ -1197,8 +1197,8 @@ module {
%91 = affine.load %7[%arg1, %arg5, %89, %90] : memref<1x512x6x6xf32>
%92 = affine.load %85[%arg2, %arg5, %arg6, %arg7] : memref<512x512x3x3xf32>
%93 = affine.load %6[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%94 = mulf %91, %92 : f32
%95 = addf %93, %94 : f32
%94 = arith.mulf %91, %92 : f32
%95 = arith.addf %93, %94 : f32
affine.store %95, %6[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
@ -1213,7 +1213,7 @@ module {
affine.for %arg4 = 0 to 4 {
%89 = affine.load %6[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%90 = affine.load %11[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%91 = addf %89, %90 : f32
%91 = arith.addf %89, %90 : f32
affine.store %91, %5[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
@ -1224,7 +1224,7 @@ module {
affine.for %arg3 = 0 to 4 {
affine.for %arg4 = 0 to 4 {
%89 = affine.load %5[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%90 = cmpf "olt", %89, %cst_0 : f32
%90 = arith.cmpf "olt", %89, %cst_0 : f32
%91 = select %90, %cst_0, %89 : f32
affine.store %91, %4[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
@ -1246,19 +1246,19 @@ module {
affine.for %arg4 = 0 to 4 {
%89 = affine.load %4[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%90 = affine.load %3[%arg1, %arg2, %c0, %c0] : memref<1x512x1x1xf32>
%91 = addf %90, %89 : f32
%91 = arith.addf %90, %89 : f32
affine.store %91, %3[%arg1, %arg2, %c0, %c0] : memref<1x512x1x1xf32>
}
}
}
}
%86 = uitofp %c16_i64 : i64 to f32
%86 = arith.uitofp %c16_i64 : i64 to f32
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 512 {
affine.for %arg3 = 0 to 1 {
affine.for %arg4 = 0 to 1 {
%89 = affine.load %3[%arg1, %arg2, %arg3, %arg4] : memref<1x512x1x1xf32>
%90 = divf %89, %86 : f32
%90 = arith.divf %89, %86 : f32
affine.store %90, %3[%arg1, %arg2, %arg3, %arg4] : memref<1x512x1x1xf32>
}
}
@ -1275,8 +1275,8 @@ module {
}
}
}
%87 = "krnl.global"() {name = "constant_20", offset = 44636928 : i64, shape = [10, 512]} : () -> memref<10x512xf32>
%88 = "krnl.global"() {name = "constant_21", offset = 44657408 : i64, shape = [10], value = dense<[-0.0278122574, -0.0410711654, -0.00870433543, -0.0383192934, -0.0207557045, 0.0408722125, 0.0364007317, 0.0409536846, 0.00303884572, -0.0432778746]> : tensor<10xf32>} : () -> memref<10xf32>
%87 = "krnl.global"() {name = "arith.constant_20", offset = 44636928 : i64, shape = [10, 512]} : () -> memref<10x512xf32>
%88 = "krnl.global"() {name = "arith.constant_21", offset = 44657408 : i64, shape = [10], value = dense<[-0.0278122574, -0.0410711654, -0.00870433543, -0.0383192934, -0.0207557045, 0.0408722125, 0.0364007317, 0.0409536846, 0.00303884572, -0.0432778746]> : tensor<10xf32>} : () -> memref<10xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 10 {
affine.store %cst_0, %1[%arg1, %arg2] : memref<1x10xf32>
@ -1284,15 +1284,15 @@ module {
%94 = affine.load %2[%arg1, %arg3] : memref<1x512xf32>
%95 = affine.load %87[%arg2, %arg3] : memref<10x512xf32>
%96 = affine.load %1[%arg1, %arg2] : memref<1x10xf32>
%97 = mulf %94, %95 : f32
%98 = addf %96, %97 : f32
%97 = arith.mulf %94, %95 : f32
%98 = arith.addf %96, %97 : f32
affine.store %98, %1[%arg1, %arg2] : memref<1x10xf32>
}
%89 = affine.load %1[%arg1, %arg2] : memref<1x10xf32>
%90 = mulf %cst, %89 : f32
%90 = arith.mulf %cst, %89 : f32
%91 = affine.load %88[%arg2] : memref<10xf32>
%92 = mulf %cst, %91 : f32
%93 = addf %90, %92 : f32
%92 = arith.mulf %cst, %91 : f32
%93 = arith.addf %90, %92 : f32
affine.store %93, %1[%arg1, %arg2] : memref<1x10xf32>
}
}

136
samples/onnx-mlir/vgg16/vgg16.mlir
View File

@ -24,12 +24,12 @@
module {
%0 = "krnl.packed_const"() {file_name = "/tmp/packed_const-708c58.tmp", is_le = true, size_in_bytes = 58862336 : i64} : () -> i64
func @main_graph(%arg0: memref<1x3x32x32xf32>) -> memref<1x10xf32> attributes {input_names = ["input.1"], output_names = ["44"]} {
%c0 = constant 0 : index
%c1_i64 = constant 1 : i64
%c512 = constant 512 : index
%c1 = constant 1 : index
%cst = constant 1.000000e+00 : f32
%cst_0 = constant 0.000000e+00 : f32
%c0 = arith.constant 0 : index
%c1_i64 = arith.constant 1 : i64
%c512 = arith.constant 512 : index
%c1 = arith.constant 1 : index
%cst = arith.constant 1.000000e+00 : f32
%cst_0 = arith.constant 0.000000e+00 : f32
%1 = memref.alloc() : memref<1x10xf32>
%2 = memref.alloc() : memref<1x512xf32>
%3 = memref.alloc() : memref<1x512x1x1xf32>
@ -72,7 +72,7 @@ module {
%40 = memref.alloc() : memref<1x64x32x32xf32>
%41 = memref.alloc() : memref<1x64x32x32xf32>
%42 = memref.alloc() : memref<1x3x34x34xf32>
%43 = "krnl.global"() {name = "constant_0", offset = 0 : i64, shape = [64, 3, 3, 3]} : () -> memref<64x3x3x3xf32>
%43 = "krnl.global"() {name = "arith.constant_0", offset = 0 : i64, shape = [64, 3, 3, 3]} : () -> memref<64x3x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 3 {
affine.for %arg3 = 0 to 34 {
@ -107,8 +107,8 @@ module {
%61 = affine.load %42[%arg1, %arg5, %59, %60] : memref<1x3x34x34xf32>
%62 = affine.load %43[%arg2, %arg5, %arg6, %arg7] : memref<64x3x3x3xf32>
%63 = affine.load %41[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
%64 = mulf %61, %62 : f32
%65 = addf %63, %64 : f32
%64 = arith.mulf %61, %62 : f32
%65 = arith.addf %63, %64 : f32
affine.store %65, %41[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
}
}
@ -122,14 +122,14 @@ module {
affine.for %arg3 = 0 to 32 {
affine.for %arg4 = 0 to 32 {
%59 = affine.load %41[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
%60 = cmpf "olt", %59, %cst_0 : f32
%60 = arith.cmpf "olt", %59, %cst_0 : f32
%61 = select %60, %cst_0, %59 : f32
affine.store %61, %40[%arg1, %arg2, %arg3, %arg4] : memref<1x64x32x32xf32>
}
}
}
}
%44 = "krnl.global"() {name = "constant_1", offset = 6912 : i64, shape = [64, 64, 3, 3]} : () -> memref<64x64x3x3xf32>
%44 = "krnl.global"() {name = "arith.constant_1", offset = 6912 : i64, shape = [64, 64, 3, 3]} : () -> memref<64x64x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 64 {
affine.for %arg3 = 0 to 34 {
@ -164,8 +164,8 @@ module {
%61 = affine.load %39[%arg1, %arg5, %59, %60] : memref<1x64x34x34xf32>
%62 = affine.load %44[%arg2, %arg5, %arg6, %arg7] : memref<64x64x3x3xf32>
%63 = affine.load %38[%arg1, %arg2, %arg3, %arg4] : memref<1x64x16x16xf32>
%64 = mulf %61, %62 : f32
%65 = addf %63, %64 : f32
%64 = arith.mulf %61, %62 : f32
%65 = arith.addf %63, %64 : f32
affine.store %65, %38[%arg1, %arg2, %arg3, %arg4] : memref<1x64x16x16xf32>
}
}
@ -179,14 +179,14 @@ module {
affine.for %arg3 = 0 to 16 {
affine.for %arg4 = 0 to 16 {
%59 = affine.load %38[%arg1, %arg2, %arg3, %arg4] : memref<1x64x16x16xf32>
%60 = cmpf "olt", %59, %cst_0 : f32
%60 = arith.cmpf "olt", %59, %cst_0 : f32
%61 = select %60, %cst_0, %59 : f32
affine.store %61, %37[%arg1, %arg2, %arg3, %arg4] : memref<1x64x16x16xf32>
}
}
}
}
%45 = "krnl.global"() {name = "constant_2", offset = 154368 : i64, shape = [128, 64, 3, 3]} : () -> memref<128x64x3x3xf32>
%45 = "krnl.global"() {name = "arith.constant_2", offset = 154368 : i64, shape = [128, 64, 3, 3]} : () -> memref<128x64x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 64 {
affine.for %arg3 = 0 to 18 {
@ -221,8 +221,8 @@ module {
%61 = affine.load %36[%arg1, %arg5, %59, %60] : memref<1x64x18x18xf32>
%62 = affine.load %45[%arg2, %arg5, %arg6, %arg7] : memref<128x64x3x3xf32>
%63 = affine.load %35[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
%64 = mulf %61, %62 : f32
%65 = addf %63, %64 : f32
%64 = arith.mulf %61, %62 : f32
%65 = arith.addf %63, %64 : f32
affine.store %65, %35[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
}
}
@ -236,14 +236,14 @@ module {
affine.for %arg3 = 0 to 16 {
affine.for %arg4 = 0 to 16 {
%59 = affine.load %35[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
%60 = cmpf "olt", %59, %cst_0 : f32
%60 = arith.cmpf "olt", %59, %cst_0 : f32
%61 = select %60, %cst_0, %59 : f32
affine.store %61, %34[%arg1, %arg2, %arg3, %arg4] : memref<1x128x16x16xf32>
}
}
}
}
%46 = "krnl.global"() {name = "constant_3", offset = 449280 : i64, shape = [128, 128, 3, 3]} : () -> memref<128x128x3x3xf32>
%46 = "krnl.global"() {name = "arith.constant_3", offset = 449280 : i64, shape = [128, 128, 3, 3]} : () -> memref<128x128x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 128 {
affine.for %arg3 = 0 to 18 {
@ -278,8 +278,8 @@ module {
%61 = affine.load %33[%arg1, %arg5, %59, %60] : memref<1x128x18x18xf32>
%62 = affine.load %46[%arg2, %arg5, %arg6, %arg7] : memref<128x128x3x3xf32>
%63 = affine.load %32[%arg1, %arg2, %arg3, %arg4] : memref<1x128x8x8xf32>
%64 = mulf %61, %62 : f32
%65 = addf %63, %64 : f32
%64 = arith.mulf %61, %62 : f32
%65 = arith.addf %63, %64 : f32
affine.store %65, %32[%arg1, %arg2, %arg3, %arg4] : memref<1x128x8x8xf32>
}
}
@ -293,14 +293,14 @@ module {
affine.for %arg3 = 0 to 8 {
affine.for %arg4 = 0 to 8 {
%59 = affine.load %32[%arg1, %arg2, %arg3, %arg4] : memref<1x128x8x8xf32>
%60 = cmpf "olt", %59, %cst_0 : f32
%60 = arith.cmpf "olt", %59, %cst_0 : f32
%61 = select %60, %cst_0, %59 : f32
affine.store %61, %31[%arg1, %arg2, %arg3, %arg4] : memref<1x128x8x8xf32>
}
}
}
}
%47 = "krnl.global"() {name = "constant_4", offset = 1039104 : i64, shape = [256, 128, 3, 3]} : () -> memref<256x128x3x3xf32>
%47 = "krnl.global"() {name = "arith.constant_4", offset = 1039104 : i64, shape = [256, 128, 3, 3]} : () -> memref<256x128x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 128 {
affine.for %arg3 = 0 to 10 {
@ -335,8 +335,8 @@ module {
%61 = affine.load %30[%arg1, %arg5, %59, %60] : memref<1x128x10x10xf32>
%62 = affine.load %47[%arg2, %arg5, %arg6, %arg7] : memref<256x128x3x3xf32>
%63 = affine.load %29[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
%64 = mulf %61, %62 : f32
%65 = addf %63, %64 : f32
%64 = arith.mulf %61, %62 : f32
%65 = arith.addf %63, %64 : f32
affine.store %65, %29[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
}
}
@ -350,14 +350,14 @@ module {
affine.for %arg3 = 0 to 8 {
affine.for %arg4 = 0 to 8 {
%59 = affine.load %29[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
%60 = cmpf "olt", %59, %cst_0 : f32
%60 = arith.cmpf "olt", %59, %cst_0 : f32
%61 = select %60, %cst_0, %59 : f32
affine.store %61, %28[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
}
}
}
}
%48 = "krnl.global"() {name = "constant_5", offset = 2218752 : i64, shape = [256, 256, 3, 3]} : () -> memref<256x256x3x3xf32>
%48 = "krnl.global"() {name = "arith.constant_5", offset = 2218752 : i64, shape = [256, 256, 3, 3]} : () -> memref<256x256x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 256 {
affine.for %arg3 = 0 to 10 {
@ -392,8 +392,8 @@ module {
%61 = affine.load %27[%arg1, %arg5, %59, %60] : memref<1x256x10x10xf32>
%62 = affine.load %48[%arg2, %arg5, %arg6, %arg7] : memref<256x256x3x3xf32>
%63 = affine.load %26[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
%64 = mulf %61, %62 : f32
%65 = addf %63, %64 : f32
%64 = arith.mulf %61, %62 : f32
%65 = arith.addf %63, %64 : f32
affine.store %65, %26[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
}
}
@ -407,14 +407,14 @@ module {
affine.for %arg3 = 0 to 8 {
affine.for %arg4 = 0 to 8 {
%59 = affine.load %26[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
%60 = cmpf "olt", %59, %cst_0 : f32
%60 = arith.cmpf "olt", %59, %cst_0 : f32
%61 = select %60, %cst_0, %59 : f32
affine.store %61, %25[%arg1, %arg2, %arg3, %arg4] : memref<1x256x8x8xf32>
}
}
}
}
%49 = "krnl.global"() {name = "constant_6", offset = 4578048 : i64, shape = [256, 256, 3, 3]} : () -> memref<256x256x3x3xf32>
%49 = "krnl.global"() {name = "arith.constant_6", offset = 4578048 : i64, shape = [256, 256, 3, 3]} : () -> memref<256x256x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 256 {
affine.for %arg3 = 0 to 10 {
@ -449,8 +449,8 @@ module {
%61 = affine.load %24[%arg1, %arg5, %59, %60] : memref<1x256x10x10xf32>
%62 = affine.load %49[%arg2, %arg5, %arg6, %arg7] : memref<256x256x3x3xf32>
%63 = affine.load %23[%arg1, %arg2, %arg3, %arg4] : memref<1x256x4x4xf32>
%64 = mulf %61, %62 : f32
%65 = addf %63, %64 : f32
%64 = arith.mulf %61, %62 : f32
%65 = arith.addf %63, %64 : f32
affine.store %65, %23[%arg1, %arg2, %arg3, %arg4] : memref<1x256x4x4xf32>
}
}
@ -464,14 +464,14 @@ module {
affine.for %arg3 = 0 to 4 {
affine.for %arg4 = 0 to 4 {
%59 = affine.load %23[%arg1, %arg2, %arg3, %arg4] : memref<1x256x4x4xf32>
%60 = cmpf "olt", %59, %cst_0 : f32
%60 = arith.cmpf "olt", %59, %cst_0 : f32
%61 = select %60, %cst_0, %59 : f32
affine.store %61, %22[%arg1, %arg2, %arg3, %arg4] : memref<1x256x4x4xf32>
}
}
}
}
%50 = "krnl.global"() {name = "constant_7", offset = 6937344 : i64, shape = [512, 256, 3, 3]} : () -> memref<512x256x3x3xf32>
%50 = "krnl.global"() {name = "arith.constant_7", offset = 6937344 : i64, shape = [512, 256, 3, 3]} : () -> memref<512x256x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 256 {
affine.for %arg3 = 0 to 6 {
@ -506,8 +506,8 @@ module {
%61 = affine.load %21[%arg1, %arg5, %59, %60] : memref<1x256x6x6xf32>
%62 = affine.load %50[%arg2, %arg5, %arg6, %arg7] : memref<512x256x3x3xf32>
%63 = affine.load %20[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%64 = mulf %61, %62 : f32
%65 = addf %63, %64 : f32
%64 = arith.mulf %61, %62 : f32
%65 = arith.addf %63, %64 : f32
affine.store %65, %20[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
@ -521,14 +521,14 @@ module {
affine.for %arg3 = 0 to 4 {
affine.for %arg4 = 0 to 4 {
%59 = affine.load %20[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%60 = cmpf "olt", %59, %cst_0 : f32
%60 = arith.cmpf "olt", %59, %cst_0 : f32
%61 = select %60, %cst_0, %59 : f32
affine.store %61, %19[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
}
}
%51 = "krnl.global"() {name = "constant_8", offset = 11655936 : i64, shape = [512, 512, 3, 3]} : () -> memref<512x512x3x3xf32>
%51 = "krnl.global"() {name = "arith.constant_8", offset = 11655936 : i64, shape = [512, 512, 3, 3]} : () -> memref<512x512x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 512 {
affine.for %arg3 = 0 to 6 {
@ -563,8 +563,8 @@ module {
%61 = affine.load %18[%arg1, %arg5, %59, %60] : memref<1x512x6x6xf32>
%62 = affine.load %51[%arg2, %arg5, %arg6, %arg7] : memref<512x512x3x3xf32>
%63 = affine.load %17[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%64 = mulf %61, %62 : f32
%65 = addf %63, %64 : f32
%64 = arith.mulf %61, %62 : f32
%65 = arith.addf %63, %64 : f32
affine.store %65, %17[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
@ -578,14 +578,14 @@ module {
affine.for %arg3 = 0 to 4 {
affine.for %arg4 = 0 to 4 {
%59 = affine.load %17[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
%60 = cmpf "olt", %59, %cst_0 : f32
%60 = arith.cmpf "olt", %59, %cst_0 : f32
%61 = select %60, %cst_0, %59 : f32
affine.store %61, %16[%arg1, %arg2, %arg3, %arg4] : memref<1x512x4x4xf32>
}
}
}
}
%52 = "krnl.global"() {name = "constant_9", offset = 21093120 : i64, shape = [512, 512, 3, 3]} : () -> memref<512x512x3x3xf32>
%52 = "krnl.global"() {name = "arith.constant_9", offset = 21093120 : i64, shape = [512, 512, 3, 3]} : () -> memref<512x512x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 512 {
affine.for %arg3 = 0 to 6 {
@ -620,8 +620,8 @@ module {
%61 = affine.load %15[%arg1, %arg5, %59, %60] : memref<1x512x6x6xf32>
%62 = affine.load %52[%arg2, %arg5, %arg6, %arg7] : memref<512x512x3x3xf32>
%63 = affine.load %14[%arg1, %arg2, %arg3, %arg4] : memref<1x512x2x2xf32>
%64 = mulf %61, %62 : f32
%65 = addf %63, %64 : f32
%64 = arith.mulf %61, %62 : f32
%65 = arith.addf %63, %64 : f32
affine.store %65, %14[%arg1, %arg2, %arg3, %arg4] : memref<1x512x2x2xf32>
}
}
@ -635,14 +635,14 @@ module {
affine.for %arg3 = 0 to 2 {
affine.for %arg4 = 0 to 2 {
%59 = affine.load %14[%arg1, %arg2, %arg3, %arg4] : memref<1x512x2x2xf32>
%60 = cmpf "olt", %59, %cst_0 : f32
%60 = arith.cmpf "olt", %59, %cst_0 : f32
%61 = select %60, %cst_0, %59 : f32
affine.store %61, %13[%arg1, %arg2, %arg3, %arg4] : memref<1x512x2x2xf32>
}
}
}
}
%53 = "krnl.global"() {name = "constant_10", offset = 30530304 : i64, shape = [512, 512, 3, 3]} : () -> memref<512x512x3x3xf32>
%53 = "krnl.global"() {name = "arith.constant_10", offset = 30530304 : i64, shape = [512, 512, 3, 3]} : () -> memref<512x512x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 512 {
affine.for %arg3 = 0 to 4 {
@ -677,8 +677,8 @@ module {
%61 = affine.load %12[%arg1, %arg5, %59, %60] : memref<1x512x4x4xf32>
%62 = affine.load %53[%arg2, %arg5, %arg6, %arg7] : memref<512x512x3x3xf32>
%63 = affine.load %11[%arg1, %arg2, %arg3, %arg4] : memref<1x512x2x2xf32>
%64 = mulf %61, %62 : f32
%65 = addf %63, %64 : f32
%64 = arith.mulf %61, %62 : f32
%65 = arith.addf %63, %64 : f32
affine.store %65, %11[%arg1, %arg2, %arg3, %arg4] : memref<1x512x2x2xf32>
}
}
@ -692,14 +692,14 @@ module {
affine.for %arg3 = 0 to 2 {
affine.for %arg4 = 0 to 2 {
%59 = affine.load %11[%arg1, %arg2, %arg3, %arg4] : memref<1x512x2x2xf32>
%60 = cmpf "olt", %59, %cst_0 : f32
%60 = arith.cmpf "olt", %59, %cst_0 : f32
%61 = select %60, %cst_0, %59 : f32
affine.store %61, %10[%arg1, %arg2, %arg3, %arg4] : memref<1x512x2x2xf32>
}
}
}
}
%54 = "krnl.global"() {name = "constant_11", offset = 39967488 : i64, shape = [512, 512, 3, 3]} : () -> memref<512x512x3x3xf32>
%54 = "krnl.global"() {name = "arith.constant_11", offset = 39967488 : i64, shape = [512, 512, 3, 3]} : () -> memref<512x512x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 512 {
affine.for %arg3 = 0 to 4 {
@ -734,8 +734,8 @@ module {
%61 = affine.load %9[%arg1, %arg5, %59, %60] : memref<1x512x4x4xf32>
%62 = affine.load %54[%arg2, %arg5, %arg6, %arg7] : memref<512x512x3x3xf32>
%63 = affine.load %8[%arg1, %arg2, %arg3, %arg4] : memref<1x512x2x2xf32>
%64 = mulf %61, %62 : f32
%65 = addf %63, %64 : f32
%64 = arith.mulf %61, %62 : f32
%65 = arith.addf %63, %64 : f32
affine.store %65, %8[%arg1, %arg2, %arg3, %arg4] : memref<1x512x2x2xf32>
}
}
@ -749,14 +749,14 @@ module {
affine.for %arg3 = 0 to 2 {
affine.for %arg4 = 0 to 2 {
%59 = affine.load %8[%arg1, %arg2, %arg3, %arg4] : memref<1x512x2x2xf32>
%60 = cmpf "olt", %59, %cst_0 : f32
%60 = arith.cmpf "olt", %59, %cst_0 : f32
%61 = select %60, %cst_0, %59 : f32
affine.store %61, %7[%arg1, %arg2, %arg3, %arg4] : memref<1x512x2x2xf32>
}
}
}
}
%55 = "krnl.global"() {name = "constant_12", offset = 49404672 : i64, shape = [512, 512, 3, 3]} : () -> memref<512x512x3x3xf32>
%55 = "krnl.global"() {name = "arith.constant_12", offset = 49404672 : i64, shape = [512, 512, 3, 3]} : () -> memref<512x512x3x3xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 512 {
affine.for %arg3 = 0 to 4 {
@ -791,8 +791,8 @@ module {
%61 = affine.load %6[%arg1, %arg5, %59, %60] : memref<1x512x4x4xf32>
%62 = affine.load %55[%arg2, %arg5, %arg6, %arg7] : memref<512x512x3x3xf32>
%63 = affine.load %5[%arg1, %arg2, %arg3, %arg4] : memref<1x512x1x1xf32>
%64 = mulf %61, %62 : f32
%65 = addf %63, %64 : f32
%64 = arith.mulf %61, %62 : f32
%65 = arith.addf %63, %64 : f32
affine.store %65, %5[%arg1, %arg2, %arg3, %arg4] : memref<1x512x1x1xf32>
}
}
@ -806,7 +806,7 @@ module {
affine.for %arg3 = 0 to 1 {
affine.for %arg4 = 0 to 1 {
%59 = affine.load %5[%arg1, %arg2, %arg3, %arg4] : memref<1x512x1x1xf32>
%60 = cmpf "olt", %59, %cst_0 : f32
%60 = arith.cmpf "olt", %59, %cst_0 : f32
%61 = select %60, %cst_0, %59 : f32
affine.store %61, %4[%arg1, %arg2, %arg3, %arg4] : memref<1x512x1x1xf32>
}
@ -828,19 +828,19 @@ module {
affine.for %arg4 = 0 to 1 {
%59 = affine.load %4[%arg1, %arg2, %arg3, %arg4] : memref<1x512x1x1xf32>
%60 = affine.load %3[%arg1, %arg2, %c0, %c0] : memref<1x512x1x1xf32>
%61 = addf %60, %59 : f32
%61 = arith.addf %60, %59 : f32
affine.store %61, %3[%arg1, %arg2, %c0, %c0] : memref<1x512x1x1xf32>
}
}
}
}
%56 = uitofp %c1_i64 : i64 to f32
%56 = arith.uitofp %c1_i64 : i64 to f32
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 512 {
affine.for %arg3 = 0 to 1 {
affine.for %arg4 = 0 to 1 {
%59 = affine.load %3[%arg1, %arg2, %arg3, %arg4] : memref<1x512x1x1xf32>
%60 = divf %59, %56 : f32
%60 = arith.divf %59, %56 : f32
affine.store %60, %3[%arg1, %arg2, %arg3, %arg4] : memref<1x512x1x1xf32>
}
}
@ -857,8 +857,8 @@ module {
}
}
}
%57 = "krnl.global"() {name = "constant_13", offset = 58841856 : i64, shape = [10, 512]} : () -> memref<10x512xf32>
%58 = "krnl.global"() {name = "constant_14", offset = 58862336 : i64, shape = [10], value = dense<[-0.0237901043, -0.0245888866, -0.0121503044, 0.0312467664, -0.0380312204, 0.0233059153, -0.0163965411, 0.0233793724, -0.0315366313, 0.0359465145]> : tensor<10xf32>} : () -> memref<10xf32>
%57 = "krnl.global"() {name = "arith.constant_13", offset = 58841856 : i64, shape = [10, 512]} : () -> memref<10x512xf32>
%58 = "krnl.global"() {name = "arith.constant_14", offset = 58862336 : i64, shape = [10], value = dense<[-0.0237901043, -0.0245888866, -0.0121503044, 0.0312467664, -0.0380312204, 0.0233059153, -0.0163965411, 0.0233793724, -0.0315366313, 0.0359465145]> : tensor<10xf32>} : () -> memref<10xf32>
affine.for %arg1 = 0 to 1 {
affine.for %arg2 = 0 to 10 {
affine.store %cst_0, %1[%arg1, %arg2] : memref<1x10xf32>
@ -866,15 +866,15 @@ module {
%64 = affine.load %2[%arg1, %arg3] : memref<1x512xf32>
%65 = affine.load %57[%arg2, %arg3] : memref<10x512xf32>
%66 = affine.load %1[%arg1, %arg2] : memref<1x10xf32>
%67 = mulf %64, %65 : f32
%68 = addf %66, %67 : f32
%67 = arith.mulf %64, %65 : f32
%68 = arith.addf %66, %67 : f32
affine.store %68, %1[%arg1, %arg2] : memref<1x10xf32>
}
%59 = affine.load %1[%arg1, %arg2] : memref<1x10xf32>
%60 = mulf %cst, %59 : f32
%60 = arith.mulf %cst, %59 : f32
%61 = affine.load %58[%arg2] : memref<10xf32>
%62 = mulf %cst, %61 : f32
%63 = addf %60, %62 : f32
%62 = arith.mulf %cst, %61 : f32
%63 = arith.addf %60, %62 : f32
affine.store %63, %1[%arg1, %arg2] : memref<1x10xf32>
}
}