From 6445563268fc4e7af9924dde1efa23f87eac6e51 Mon Sep 17 00:00:00 2001
From: lucasdelimanogueira <lucasln96@hotmail.com>
Date: Mon, 6 May 2024 10:24:43 -0300
Subject: [PATCH] Fix radd rsub

---
 norch/__pycache__/tensor.cpython-38.pyc | Bin 11598 -> 12248 bytes
 norch/tensor.py                         |  50 +++++++++++++++++++++++-
 test.py                                 |   2 +-
 3 files changed, 50 insertions(+), 2 deletions(-)

diff --git a/norch/__pycache__/tensor.cpython-38.pyc b/norch/__pycache__/tensor.cpython-38.pyc
index e2301213bdd2c7325719502d64a73d2f9a5cfe38..77de118802c1ba06a14303e163467aefbfb2a604 100644
GIT binary patch
delta 747
zcmZ{gJxo(k6vy9rO--TnJ+!Z_(3*ZwD5dqI0V9aEE+h_4mXFecH^421N@;y<6=D-j
zsK(96S<uD7n9xo07DJ*&U36hoaWFB)UBl$Z;61hy7Vq%8x#!;h|9|e;N`GL=n9bHI
ziM72wIRD`M6XiQVU`-x`(>Nsi?Qw47N)6H~S9yC)O1JX_=H>MOKfwpZG02DbNwIfl
zd-yOvRg+dB9pZ9Q3ZF4trj`9r$Gch#>c(qr2F#FCsuISvkq)Rzt8&%Ew5lm7=}8f<
zj}j)(e^_f_OrPPLu`gI~gcWr(yHqOZM;cZ9?e^e1`=)TNItP@xL+>Hs7-1ZrJM%Ds
zFBLl`To(8x^!<3v<wVufB1Cil?I1()sof4*MOLXY%a|Ro=;+<sf+%A#;4$vI-h;J~
zN-_twhYsRbO~r3rX58qwXH8K#MmYAjA=YV;BY+E?*KM<+8kY4SU8es1IA2+0EQtr)
z=U@(*+Xv%#!#xL+f)3+{J78*l?5NiYQ^v4Y1@YpuW*31jtoTNn*IPadtYIS1g+F~S
z#EVV;wDl4Nc?rAoEf%ga_Q*&C)?r_Bb`7s`(`Px@22UT}3{uLbzYiaG_gk0t_=p)R
zZDLGiy^`e%R~Y+bs6A%^c8pZ$ILKz%xYPR<WS5NTzE_aNLy;WB@OESXdhuCgIWtLK
zX~Hz&0)ghDbHY5~A|XduB;*MNLXmKppc6_0A1y__P|WJ(6}^nNqJD_uMl^_TqJ2=o
HooMC{>*UbV

delta 546
zcmYk&KWGzC90&0GU2AMT?~>+E6ZD!i=6cO(Ob~0Ew1TUs6r@ce(RgXOv|3FQPpft)
z4$UAa4t|%eBDjb`xuHlwp-wJ>_^09`4&vO|LEp~^`i9T&-k0C^z4zIEThWTDx<?W#
zzn!0Za(Z3cAqqZ}^He}VPKL%E)zR8i@jHGe(3UCzX95j*Em$|4F(==aR;e=N$WxMa
z+#Bv%?jsA2!biyR-iK!h3(>OJ*ouy5R>U<$hdaegqdV51BHn2s-)`H!zBG&V{tPx^
zgZO3idj0#hN$CVR9l?DohJgNzN_eRc$v-80*YBto#F-_#21iP`HlR}q4+hRq6({2R
zeT(&Z+cr^)N6;6a6!)pc$GSR|c$!ow<837UMAO&_hmg>B;p>5N=eSqgsMWk>+q2j-
zQn;CVB~qhCRjG4Rf+>o+sP9i`$=fv6$oKbI|AqyzYF5R-Z)Sq-KBi%=8Z{0glRm0k
z`frf<0<5G}dhXpzpCo$ieXx#^{E<B6&>NC}5I$wL$i_s^AYL3kMTG9`2$?8kZ=Wsm
zm~#w&g?o{yF*A(A%rW!KWoCi7!dzt<f-gNSx7n(@H<sN+?9C-<9NC<KQqH6%R&wWm
E0|pzD;Q#;t

diff --git a/norch/tensor.py b/norch/tensor.py
index 633b8ae..91b6e2f 100644
--- a/norch/tensor.py
+++ b/norch/tensor.py
@@ -224,10 +224,34 @@ class Tensor:
 
         return result_data
     
+    def __radd__(self, other):
+        if isinstance(other, (int, float)):
+            other = other * self.ones_like()
+        
+        if self.shape != other.shape:
+            raise ValueError("Tensors must have the same shape for addition")
+        
+        Tensor._C.add_tensor.argtypes = [ctypes.POINTER(CTensor), ctypes.POINTER(CTensor)]
+        Tensor._C.add_tensor.restype = ctypes.POINTER(CTensor)
+
+        result_tensor_ptr = Tensor._C.add_tensor(other.tensor, self.tensor)
+
+        result_data = Tensor()
+        result_data.tensor = result_tensor_ptr
+        result_data.shape = self.shape.copy()
+        result_data.ndim = self.ndim
+        result_data.device = self.device
+        
+        result_data.requires_grad = self.requires_grad or other.requires_grad
+        if result_data.requires_grad:
+            result_data.grad_fn = AddBackward(other, self)
+
+        return result_data
+    
     def __sub__(self, other):
         if isinstance(other, (int, float)):
             other = other * self.ones_like()
-            
+
         if self.shape != other.shape:
             raise ValueError("Tensors must have the same shape for subtraction")
         
@@ -248,6 +272,30 @@ class Tensor:
 
         return result_data
     
+    def __rsub__(self, other):
+        if isinstance(other, (int, float)):
+            other = other * self.ones_like()
+
+        if self.shape != other.shape:
+            raise ValueError("Tensors must have the same shape for subtraction")
+        
+        Tensor._C.sub_tensor.argtypes = [ctypes.POINTER(CTensor), ctypes.POINTER(CTensor)]
+        Tensor._C.sub_tensor.restype = ctypes.POINTER(CTensor)
+
+        result_tensor_ptr = Tensor._C.sub_tensor(other.tensor, self.tensor)
+
+        result_data = Tensor()
+        result_data.tensor = result_tensor_ptr
+        result_data.shape = self.shape.copy()
+        result_data.ndim = self.ndim
+        result_data.device = self.device
+
+        result_data.requires_grad = self.requires_grad or other.requires_grad
+        if result_data.requires_grad:
+            result_data.grad_fn = SubBackward(other, self)
+
+        return result_data
+    
     def __mul__(self, other):
         if isinstance(other, (int, float)):
             result_data = Tensor()
diff --git a/test.py b/test.py
index e473a71..a490bd6 100644
--- a/test.py
+++ b/test.py
@@ -75,7 +75,7 @@ if __name__ == "__main__":
                         [[19, 20], [21, 22], [23, 24]],
                         [[25, 26], [27, 28], [29, 30]]], requires_grad=True)
     
-    result = 2 ** tensor1
+    result =  (-10) - tensor1
     result = result.sum()
     result.backward()
     print(tensor1.grad)