Fix unsigned integers for collapse/expand

The `tensor.expand` and `tensor.collapse` operations need to have their signedness cleaned up.
rsuderman · Oct 26, 2023 · 5f7030d · 5f7030d
1 parent 41c2b6d
commit 5f7030d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/compiler/src/iree/compiler/InputConversion/StableHLO/StableHLOToIREEInputDialects.cpp b/compiler/src/iree/compiler/InputConversion/StableHLO/StableHLOToIREEInputDialects.cpp
@@ -529,6 +529,8 @@ struct ConvertStableHloToIreeInputDialects final
           ml_program::GlobalLoadConstOp::getOperationName(),
           ml_program::GlobalStoreOp::getOperationName(),
           tensor::FromElementsOp::getOperationName(),
+          tensor::CollapseShapeOp::getOperationName(),
+          tensor::ExpandShapeOp::getOperationName(),
           arith::IndexCastUIOp::getOperationName()}) {
       patterns.add<GenericTypeConvert>(opName, *typeConverter, context);
     }