Implement the inplace logic in src/ntops/torch.py

Author: stever178

src/ntops/kernels/relu.py

@@ -6,22 +6,12 @@
 from ntops.kernels.element_wise import arrangement
 
 
-def default_application(input, output):
+def application(input, output):
     output = max(0.0, input)  # noqa: F841
 
 
-def True_application(input, output):
-    input = max(0.0, input)  
-    output = input  # noqa: F841
-
-
 @functools.cache
-def make(ndim, replace):
-    if replace == True:
-        application = True_application
-    else:
-        application = default_application
-
+def make(ndim):
     tensors = (Tensor(ndim), Tensor(ndim))
 
     return ninetoothed.make(arrangement, application, tensors)

src/ntops/torch.py

@@ -121,9 +121,12 @@ def mul(input, other, *, out=None):
 
 
 def relu(input, inplace=False):
-    output = torch.empty_like(input)
+    if inplace:
+        output = input
+    else:
+        output = torch.empty_like(input)
 
-    kernel = ntops.kernels.relu.make(input.ndim, inplace)
+    kernel = ntops.kernels.relu.make(input.ndim)
 
     kernel(input, output)