Remove not using tensor for backward

trident/operation/linear.py

@@ -29,20 +29,18 @@ def forward(ctx: Any, *args: Any, **kwargs: Any):
         output = Linear.__forward(input, weight, bias, use_accelerator)
         util.pop_trace()
 
-        ctx.save_for_backward(input, weight, bias, output)
+        ctx.save_for_backward(input, weight, bias)
         ctx.use_accelerator = use_accelerator
 
         return output
 
     @staticmethod
     def backward(ctx: Any, *grad_outputs: Any):
         (grad_output,) = grad_outputs
-        input, weight, bias, output = ctx.saved_tensors
+        input, weight, bias = ctx.saved_tensors
 
         util.push_trace("Linear.__backward")
-        grad_input, grad_weight, grad_bias = Linear.__backward(
-            grad_output, output, input, weight, bias, ctx.use_accelerator
-        )
+        grad_input, grad_weight, grad_bias = Linear.__backward(grad_output, input, weight, bias, ctx.use_accelerator)
         util.pop_trace()
 
         return grad_input, grad_weight, grad_bias, None, None
@@ -81,7 +79,7 @@ def grid(meta):
         return output
 
     @staticmethod
-    def __backward(grad_output, output, input, weight, bias, use_accelerator):
+    def __backward(grad_output, input, weight, bias, use_accelerator):
         factory_kwargs = {"device": input.device, "dtype": input.dtype}
         num_batches, m_size, k_size = input.shape
         n_size, _ = weight.shape