[LLVMCPU] Fix llvmcpu check before conversion for complex types (iree-org#21644)

This PR fixes  `LLVMCPUCheckIRBeforeLLVMConversionPass` .

The pass erroneously assumed that the tensor element type is an integer
or a float and would therefore fail for complex types.

A test was added to cover this case.

---------

Signed-off-by: Giacomo Castiglioni <[email protected]>
Signed-off-by: keshavvinayak01 <[email protected]>

Author: castigli

compiler/src/iree/compiler/Codegen/LLVMCPU/LLVMCPUCheckIRBeforeLLVMConversion.cpp

@@ -93,7 +93,7 @@ checkStackAllocationSize(mlir::FunctionOpInterface funcOp) {
       }
       return allocaOp.emitOpError("expected no unbounded stack allocations");
     }
-    allocaSize *= allocaType.getElementType().getIntOrFloatBitWidth();
+    allocaSize *= IREE::Util::getTypeBitWidth(allocaType.getElementType());
     if (allocaOp.getAlignment()) {
       int64_t alignmentInBits = *allocaOp.getAlignment() * 8;
       allocaSize =

compiler/src/iree/compiler/Codegen/LLVMCPU/test/check_ir_before_llvm_conversion.mlir

@@ -1,4 +1,4 @@
-// RUN: iree-opt --pass-pipeline="builtin.module(func.func(iree-llvmcpu-check-ir-before-llvm-conversion))" %s --verify-diagnostics -split-input-file
+// RUN: iree-opt --pass-pipeline="builtin.module(func.func(iree-llvmcpu-check-ir-before-llvm-conversion))" %s --verify-diagnostics -split-input-file | FileCheck %s
 
 func.func @dynamic_allocas(%arg0: index) {
   // expected-error @+1 {{expected no unbounded stack allocations}}
@@ -57,3 +57,11 @@ func.func @nested_op_alloca(%arg0 : index) {
   }
   return
 }
+
+// -----
+
+func.func @complex_alloca() {
+  %0 = memref.alloca() : memref<128xcomplex<f32>>
+  return
+}
+// CHECK-LABEL: func @complex_alloca(

compiler/src/iree/compiler/Codegen/LLVMCPU/test/check_ir_before_llvm_conversion_not_fail_unbound.mlir

@@ -1,4 +1,4 @@
-// RUN: iree-opt --pass-pipeline="builtin.module(func.func(iree-llvmcpu-check-ir-before-llvm-conversion{fail-on-out-of-bounds=false}))" %s -split-input-file
+// RUN: iree-opt --pass-pipeline="builtin.module(func.func(iree-llvmcpu-check-ir-before-llvm-conversion{fail-on-out-of-bounds=false}))" %s -split-input-file | FileCheck %s
 
 func.func @dynamic_allocas(%arg0: index) {
   %0 = memref.alloca(%arg0) : memref<?xf32>