fail for complex types

Signed-off-by: Nirvedh <[email protected]>
iree-org · Feb 14, 2025 · 331092b · 331092b
1 parent 6485474
commit 331092b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/compiler/src/iree/compiler/Codegen/Dialect/GPU/TargetUtils/ConfigUtils.cpp b/compiler/src/iree/compiler/Codegen/Dialect/GPU/TargetUtils/ConfigUtils.cpp
@@ -511,6 +511,9 @@ static unsigned getRepresentativeBitWidth(linalg::LinalgOp linalgOp) {
   std::optional<unsigned> maxBitWidth;
   auto updateElementTypeBitWidth = [&](Value v) {
     auto elementType = getElementTypeOrSelf(v);
+    if(!elementType.isIntOrFloat()){
+      return;
+    }
     unsigned bitWidth = elementType.getIntOrFloatBitWidth();
     if (maxBitWidth) {
       maxBitWidth = std::max(maxBitWidth.value(), bitWidth);