From 26d1be76aac316c2eb190d542af98af7c341b07b Mon Sep 17 00:00:00 2001 From: Abhishek Varma Date: Mon, 26 Aug 2024 08:40:18 +0000 Subject: [PATCH 1/4] IREE Bump to 26th Aug, 2024 --- sync_deps.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sync_deps.py b/sync_deps.py index 9f8444748..0d0e470c7 100644 --- a/sync_deps.py +++ b/sync_deps.py @@ -7,7 +7,7 @@ ### Update with: shark-workspace pin PINNED_VERSIONS = { - "iree": "5a48912c52f65ead960bec9bfde5a836d1b02ab2", + "iree": "60b65f30c932eaf967922785253a85a1aa14cebb", } ORIGINS = { From e1e348579ea99da7d4027f2075675d2782483078 Mon Sep 17 00:00:00 2001 From: Abhishek Varma Date: Mon, 26 Aug 2024 09:32:06 +0000 Subject: [PATCH 2/4] Rename iree-codegen-hoist-statically-bound-allocations --- .../Transforms/test/create_reference_to_allocation.mlir | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/compiler/plugins/target/AMD-AIE/iree-amd-aie/Transforms/test/create_reference_to_allocation.mlir b/compiler/plugins/target/AMD-AIE/iree-amd-aie/Transforms/test/create_reference_to_allocation.mlir index 8bd685873..9145088dd 100644 --- a/compiler/plugins/target/AMD-AIE/iree-amd-aie/Transforms/test/create_reference_to_allocation.mlir +++ b/compiler/plugins/target/AMD-AIE/iree-amd-aie/Transforms/test/create_reference_to_allocation.mlir @@ -1,4 +1,4 @@ -// RUN: iree-opt --split-input-file --pass-pipeline="builtin.module(func.func(iree-amdaie-create-reference-to-allocation, iree-hoist-statically-bound-allocations))" %s | FileCheck %s +// RUN: iree-opt --split-input-file --pass-pipeline="builtin.module(func.func(iree-amdaie-create-reference-to-allocation, iree-codegen-hoist-statically-bound-allocations))" %s | FileCheck %s // CHECK-LABEL: func.func @single_alloc // CHECK: %[[ALLOC:.+]] = memref.alloc() : memref<8x16xi32, 2 : i32> From 603bd5d7eab8379713e9f9180830b9f593d1dec9 Mon Sep 17 00:00:00 2001 From: Abhishek Varma Date: Wed, 28 Aug 2024 15:10:22 +0000 Subject: [PATCH 3/4] Conv e2e test update --- tests/samples/conv_pipeline_e2e.mlir | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/samples/conv_pipeline_e2e.mlir b/tests/samples/conv_pipeline_e2e.mlir index 7c6957017..ffe8222f8 100644 --- a/tests/samples/conv_pipeline_e2e.mlir +++ b/tests/samples/conv_pipeline_e2e.mlir @@ -24,7 +24,7 @@ func.func @conv_2d_nhwc_hwcf_q(%arg0: tensor<2x14x14x32xi8>, %arg1: tensor<3x3x3 return %2 : tensor<2x12x12x64xi32> } -// CHECK-LABEL: hal.executable.export public @conv_2d_nhwc_hwcf_q_dispatch_0_conv_2d_nhwc_hwcf_q_2x12x12x64x3x3x32_i8xi8xi32xi32xi32 +// CHECK-LABEL: hal.executable.export public @conv_2d_nhwc_hwcf_q_dispatch_0_conv_2d_nhwc_hwcf_2x12x12x64x3x3x32_i8xi8xi32 // CHECK: aie.device(npu1_4col) // CHECK: aie.shim_dma_allocation // CHECK: aie.shim_dma_allocation From 6c22541b6210add7fa9f250b7bd40caa2551eca1 Mon Sep 17 00:00:00 2001 From: Abhishek Varma Date: Thu, 29 Aug 2024 05:39:56 +0000 Subject: [PATCH 4/4] Add another pattern set for vectorization --- .../AMD-AIE/iree-amd-aie/Transforms/AMDAIEVectorization.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/compiler/plugins/target/AMD-AIE/iree-amd-aie/Transforms/AMDAIEVectorization.cpp b/compiler/plugins/target/AMD-AIE/iree-amd-aie/Transforms/AMDAIEVectorization.cpp index fa661b784..cec0c635b 100644 --- a/compiler/plugins/target/AMD-AIE/iree-amd-aie/Transforms/AMDAIEVectorization.cpp +++ b/compiler/plugins/target/AMD-AIE/iree-amd-aie/Transforms/AMDAIEVectorization.cpp @@ -100,6 +100,7 @@ void AMDAIEVectorizationPass::runOnOperation() { RewritePatternSet vectorizationPatterns(funcOp.getContext()); vector::populateVectorReductionToContractPatterns(vectorizationPatterns); + vector::populateSinkVectorOpsPatterns(vectorizationPatterns); // Including this pattern prevents broadcasting in vector.transfer_read ops vector::populateVectorTransferPermutationMapLoweringPatterns(