Don't crash when the input to aten.copy is unranked (#2307)

This can happen when the input comes from an unsupported operator
llvm · Jul 18, 2023 · 0c17997 · 0c17997
1 parent 718f53f
commit 0c17997
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/lib/Dialect/Torch/Transforms/DecomposeComplexOps.cpp b/lib/Dialect/Torch/Transforms/DecomposeComplexOps.cpp
@@ -3162,6 +3162,11 @@ class DecomposeAtenCopyOp : public OpRewritePattern<AtenCopyOp> {
       return rewriter.notifyMatchFailure(
           op, "expected result type to have a dtype");
     }
+    auto srcTy = op.getSrc().getType().cast<BaseTensorType>();
+    if (!srcTy.hasSizes() || !srcTy.hasDtype()) {
+      return rewriter.notifyMatchFailure(
+          op, "expected src type to have a known rank and dtype");
+    }
     Type resultDtype = resultType.getDtype();
     Value srcToDtype =
         convertTensorToDtype(rewriter, op.getLoc(), op.getSrc(), resultDtype);