[inductor] replace ops.div with ops.truediv (#112243)

Jiong Gong · pytorchmergebot · commit e061144aaf0b · 2023-11-01T05:50:51.000Z
Pull Request resolved: #112243 Approved by: https://github.com/lezcano ghstack dependencies: #112234
diff --git a/torch/_inductor/codegen/common.py b/torch/_inductor/codegen/common.py
@@ -384,7 +384,7 @@ def constant(value, dtype):
 
     @staticmethod
     def reciprocal(x):
-        return ops.div("1", x)
+        return ops.truediv("1", x)
 
     @staticmethod
     def square(x):
diff --git a/torch/_inductor/codegen/cpp.py b/torch/_inductor/codegen/cpp.py
@@ -451,10 +451,6 @@ def sub(a, b):
     def mul(a, b):
         return f"{a} * {b}"
 
-    @staticmethod
-    def div(a, b):
-        return f"{a} / {b}"
-
     @staticmethod
     def truediv(a, b):
         return f"{a} / {b}"
diff --git a/torch/_inductor/codegen/triton.py b/torch/_inductor/codegen/triton.py
@@ -1546,7 +1546,7 @@ def _mask_value(value, default):
                 sum_ = ops.reduction(dtype, dtype, "sum", value)
                 self.inside_reduction = False
                 rnumel = ops.index_expr(self.numels[-1], dtype)
-                mean = ops.div(sum_, rnumel)
+                mean = ops.truediv(sum_, rnumel)
 
                 self.inside_reduction = True
                 dx = ops.sub(value, mean)
diff --git a/torch/_inductor/lowering.py b/torch/_inductor/lowering.py
@@ -4036,7 +4036,9 @@ def end_index(index, out_dim, inp_dim):
     ones_loader = pad_adaptive_loader(ones_like(x))
 
     def fn(idx):
-        return ops.div(fn_sum(idx, pad_adaptive_loader(x)), fn_sum(idx, ones_loader))
+        return ops.truediv(
+            fn_sum(idx, pad_adaptive_loader(x)), fn_sum(idx, ones_loader)
+        )
 
     rv = Pointwise.create(
         device=x.get_device(),
@@ -4182,7 +4184,7 @@ def fn(idx):
 
         def fn(idx):
             # TODO(jansel): optimize to do `int(x<h)` rather than `x<h?1:0`
-            return ops.div(fn_sum(idx, x_loader), fn_sum(idx, ones_loader))
+            return ops.truediv(fn_sum(idx, x_loader), fn_sum(idx, ones_loader))
 
     rv = Pointwise.create(
         device=x.get_device(),
@@ -4738,7 +4740,7 @@ def div_prim(a, b):
         return truncdiv(a, b)
 
     def fn(*args):
-        return ops.div(*args)
+        return ops.truediv(*args)
 
     return make_pointwise(fn)(a, b)
 
