Skip to content

Commit eb452e1

Browse files
committed
Fix name
1 parent b976447 commit eb452e1

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

src/flag_gems/ops/div.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -41,17 +41,17 @@ def true_divide(A, B):
4141
return torch.tensor(A / B)
4242

4343

44-
def true_divide_out(A, B, OUT):
44+
def true_divide_out(A, B, out):
4545
logger.debug("GEMS TRUE_DIVIDE OUT")
4646
if isinstance(A, torch.Tensor) and isinstance(B, torch.Tensor):
47-
return true_div_func(A, B, out0=OUT)
47+
return true_div_func(A, B, out0=out)
4848
elif isinstance(A, torch.Tensor):
49-
return true_div_func_tensor_scalar(A, B, out0=OUT)
49+
return true_div_func_tensor_scalar(A, B, out0=out)
5050
elif isinstance(B, torch.Tensor):
51-
return true_div_func_scalar_tensor(A, B, out0=OUT)
51+
return true_div_func_scalar_tensor(A, B, out0=out)
5252
else:
5353
# Both scalar
54-
return torch.tensor(A / B) if OUT is None else OUT.fill_(A / B)
54+
return torch.tensor(A / B) if out is None else out.fill_(A / B)
5555

5656

5757
def true_divide_(A, B):

0 commit comments

Comments
 (0)