refine type of torch.roll

llvm · Aug 9, 2022 · 6a27099 · 6a27099
1 parent 8e9a7ec
commit 6a27099
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/Dialect/Torch/Transforms/RefineTypes.cpp b/lib/Dialect/Torch/Transforms/RefineTypes.cpp
@@ -657,7 +657,7 @@ void TypeAnalysis::visitOperation(Operation *op,
           AtenZero_Op, AtenIndexTensorOp, ValsemVariantAtenIndexPutImplOp,
           AtenIndexPutOp, ValsemVariantAtenCopyOp, AtenZeroOp,
           AtenIndexPutHackedTwinOp, AtenMaskedFillScalarOp, AtenFlipOp,
-          PrimAbsScalarOp, AtenNumpyTOp, AtenTriuOp>(op)) {
+          PrimAbsScalarOp, AtenNumpyTOp, AtenTriuOp, AtenRollOp>(op)) {
     return incorporateKnowledge(op->getResult(0), operands[0]->getValue());
   }