From 7222b621efb07521dfc512d7dc1d328a0b747670 Mon Sep 17 00:00:00 2001 From: Aleksandar Zecevic Date: Wed, 11 Dec 2024 14:27:57 +0000 Subject: [PATCH] Added test for get_dimension_size -> constant conversion --- .../get_dimension_size_decomposition.mlir | 10 ++++++++++ 1 file changed, 10 insertions(+) create mode 100644 test/ttmlir/Dialect/TTIR/Decomposition/get_dimension_size_decomposition.mlir diff --git a/test/ttmlir/Dialect/TTIR/Decomposition/get_dimension_size_decomposition.mlir b/test/ttmlir/Dialect/TTIR/Decomposition/get_dimension_size_decomposition.mlir new file mode 100644 index 0000000000..aea7e3eb15 --- /dev/null +++ b/test/ttmlir/Dialect/TTIR/Decomposition/get_dimension_size_decomposition.mlir @@ -0,0 +1,10 @@ +// RUN: ttmlir-opt --ttir-to-ttir-decomposition %s | FileCheck %s +module { + func.func @get_dimension_size_decomposition(%arg0: tensor<32x64x128xf32>) -> tensor<1xi32> { + // CHECK: [[VAL:%.+]] = "ttir.constant" + // CHECK-SAME: value = dense<128> : tensor<1xi32> + // CHECK: return [[VAL]] : tensor<1xi32> + %0 = "ttir.get_dimension_size"(%arg0) <{dimension = 2 : i32}> : (tensor<32x64x128xf32>) -> tensor<1xi32> + return %0 : tensor<1xi32> + } +}