Merge pull request #529 from AlexanderViand-Intel:tensor_of_tensor_fix

PiperOrigin-RevId: 615769882
google · Mar 14, 2024 · 1fccbc9 · 1fccbc9
2 parents ca5387e + 7321327
commit 1fccbc9
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/lib/Conversion/Utils.cpp b/lib/Conversion/Utils.cpp
@@ -222,9 +222,8 @@ struct ConvertFromElements
 void addTensorOfTensorConversionPatterns(TypeConverter &typeConverter,
                                          RewritePatternSet &patterns,
                                          ConversionTarget &target) {
-  target.addDynamicallyLegalDialect<tensor::TensorDialect>([&](Operation *op) {
-    return typeConverter.isLegal(op->getOperandTypes());
-  });
+  target.addDynamicallyLegalDialect<tensor::TensorDialect>(
+      [&](Operation *op) { return typeConverter.isLegal(op); });
 
   typeConverter.addConversion([&](TensorType type) -> Type {
     if (!typeConverter.isLegal(type.getElementType())) {
@@ -248,9 +247,8 @@ void addTensorOfTensorConversionPatterns(TypeConverter &typeConverter,
     return type;
   });
 
-  target.addDynamicallyLegalDialect<affine::AffineDialect>([&](Operation *op) {
-    return typeConverter.isLegal(op->getOperandTypes());
-  });
+  target.addDynamicallyLegalDialect<affine::AffineDialect>(
+      [&](Operation *op) { return typeConverter.isLegal(op); });
 
   patterns.add<ConvertAny, ConvertExtract, ConvertInsert, ConvertFromElements>(
       typeConverter, patterns.getContext());