Fix incorrect pow(x, 0) case. (pytorch#145)

lly-zero-one · Feb 12, 2020 · 8cfdd14 · 8cfdd14
1 parent 67a97bc
commit 8cfdd14
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torch/csrc/jit/tensorexpr/kernel.cpp b/torch/csrc/jit/tensorexpr/kernel.cpp
@@ -389,7 +389,7 @@ Tensor TensorExprKernel::ComputeValue(torch::jit::Value* v) {
               } else if (imm = 0.5f) {
                 return sqrt(lhs);
               } else if (imm == 0.0f) {
-                return Expr(0.0f);
+                return Expr(1.0f);
               } else if (imm == -0.5f) {
                 return rsqrt(lhs);
               } else if (imm == -1.0f) {
@@ -414,7 +414,7 @@ Tensor TensorExprKernel::ComputeValue(torch::jit::Value* v) {
                   Expr tmp = lhs * lhs;
                   return tmp * tmp;
                 } else if (imm == 0) {
-                  return Expr(0.0f);
+                  return Expr(1.0f);
                 } else if (imm == -1) {
                   return Expr(1.0f) / lhs;
                 } else if (imm == -2) {