Fix verifyLinalgCompatibleTypes which currently doesn't successfull…

…y catch `torch.tensor`. (llvm#947)
qedawkins · Jun 15, 2022 · a34dad2 · a34dad2
1 parent 77ab316
commit a34dad2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lib/Conversion/Utils/Utils.cpp b/lib/Conversion/Utils/Utils.cpp
@@ -26,6 +26,8 @@ LogicalResult verifyLinalgCompatibleTypes(Operation *op,
   // TODO: Remove this check but use a separate verification pass to verify the
   // invariants expected by later passes.
   auto isValidLinalgType = [](Type type) {
+    if (type.isa<NonValueTensorType>())
+      return false;
     auto tensor = type.dyn_cast<ValueTensorType>();
     return !tensor ||
            tensor.toBuiltinTensor().dyn_cast_or_null<RankedTensorType>();