There was an error while loading. Please reload this page.
1 parent b976447 commit eb452e1Copy full SHA for eb452e1
1 file changed
src/flag_gems/ops/div.py
@@ -41,17 +41,17 @@ def true_divide(A, B):
41
return torch.tensor(A / B)
42
43
44
-def true_divide_out(A, B, OUT):
+def true_divide_out(A, B, out):
45
logger.debug("GEMS TRUE_DIVIDE OUT")
46
if isinstance(A, torch.Tensor) and isinstance(B, torch.Tensor):
47
- return true_div_func(A, B, out0=OUT)
+ return true_div_func(A, B, out0=out)
48
elif isinstance(A, torch.Tensor):
49
- return true_div_func_tensor_scalar(A, B, out0=OUT)
+ return true_div_func_tensor_scalar(A, B, out0=out)
50
elif isinstance(B, torch.Tensor):
51
- return true_div_func_scalar_tensor(A, B, out0=OUT)
+ return true_div_func_scalar_tensor(A, B, out0=out)
52
else:
53
# Both scalar
54
- return torch.tensor(A / B) if OUT is None else OUT.fill_(A / B)
+ return torch.tensor(A / B) if out is None else out.fill_(A / B)
55
56
57
def true_divide_(A, B):
0 commit comments