chore: Rename var dtype to be explicit

Signed-off-by: Dheeraj Peri <peri.dheeraj@gmail.com>
pytorch · Aug 5, 2021 · 152b377 · 152b377
1 parent a6577a4
commit 152b377
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/core/lowering/passes/unpack_var.cpp b/core/lowering/passes/unpack_var.cpp
@@ -17,7 +17,7 @@ void UnpackVar(std::shared_ptr<torch::jit::Graph>& graph) {
       %none: None = prim::Constant()
       %false: bool = prim::Constant[value=0]()
       %0: int = prim::Constant[value=0]()
-      %dtype: int = prim::Constant[value=6]()
+      %f32_dtype: int = prim::Constant[value=6]()
       %1: int = prim::Constant[value=1]()
       %sqrd: Tensor = aten::mul(%input, %input)
       %sqrdmean: Tensor = aten::mean(%sqrd, %dims, %keepdim, %none)
@@ -27,7 +27,7 @@ void UnpackVar(std::shared_ptr<torch::jit::Graph>& graph) {
       %varout : Tensor = prim::If(%unbiased)
         block0():
           %shape: int[] = aten::size(%input)
-          %shapet: Tensor = aten::tensor(%shape, %dtype, %none, %false)
+          %shapet: Tensor = aten::tensor(%shape, %f32_dtype, %none, %false)
           %dim: int = prim::ListUnpack(%dims)
           %reduceddims: Tensor = aten::select(%shapet, %0, %dim)
           %numel: Tensor = aten::prod(%reduceddims, %dim, %keepdim, %none)