From 9f1ed4b2baed10cce7055ad1e13d08a59b753ab5 Mon Sep 17 00:00:00 2001
From: Yuanqiang Liu <liuyuanqiang.yqliu@bytedance.com>
Date: Sat, 6 May 2023 06:22:14 +0800
Subject: [PATCH] [Torch Dialect] typo fix for RefineTypes (#2087)

---
 lib/Dialect/Torch/Transforms/RefineTypes.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lib/Dialect/Torch/Transforms/RefineTypes.cpp b/lib/Dialect/Torch/Transforms/RefineTypes.cpp
index 33293273d..f95ea6e7f 100644
--- a/lib/Dialect/Torch/Transforms/RefineTypes.cpp
+++ b/lib/Dialect/Torch/Transforms/RefineTypes.cpp
@@ -859,7 +859,7 @@ void TypeAnalysis::visitOperation(Operation *op,
     result2Knowledge.dtype = self.dtype;
     incorporateKnowledge(op->getResult(0), result0Knowledge);
     incorporateKnowledge(op->getResult(1), result1Knowledge);
-    incorporateKnowledge(op->getResult(2), result1Knowledge);
+    incorporateKnowledge(op->getResult(2), result2Knowledge);
     return;
   }