From 13da54ca3967b4181c8e01683d34e893e5615b29 Mon Sep 17 00:00:00 2001
From: lucasdelimanogueira <lucasln96@hotmail.com>
Date: Sat, 18 May 2024 08:54:03 -0300
Subject: [PATCH] fix add broadcasted access memory wrong values

---
 norch/__pycache__/tensor.cpython-38.pyc | Bin 15534 -> 15623 bytes
 norch/tensor.py                         |   8 ++++++--
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/norch/__pycache__/tensor.cpython-38.pyc b/norch/__pycache__/tensor.cpython-38.pyc
index 45abd37a6e25c4ec09757980a929bbc399ecda78..6c93f168b96f7a34206c20121ae3073bdf07cbca 100644
GIT binary patch
delta 1363
zcmY+COH5p46vywmk6~as1%}rE@~|>(hK@K;Q3^z*Hd-5^CdJgWwI2ib4loQe!#9_w
zccks0KwU^Y#)S(58%-0bn(M-a#l*ygZrr$WHX0iriQT)Z@tiw|dXxM69_M`L|3BZo
zw^ue+gD(YxRSG>Hyn8Nr{?p*O2A@W&{32d%XgfZoY$+Q`5==1Fb$>y1R7Xka4y5#=
zdPUt*3Yw$IpYeX{(3Ecbi%P+8bjJ|EO@*l2M>mxs(5#wH)zDaDYQ$Q{Na<&lS3(7!
zW9Uk%*72QIR7KRze?)nq+_zLeM0xb4Lh>iO&QTmFs<cy|*(R+feE)&2wg2g{tDVMO
zPnG7?lzXU2Yf>ats=W=lOaHCD4N}@o|JHYm3rbXfep2NdId}{jee~ARn}3hM>9aV}
z_S`czT?bpEW(KaP1+AbvnxoHZ@|vTWsyFK8(bi0@2sdC;+0ku<vx@32fHME`rh@ZL
z7uTXj$(Nr_<gDI<v9DAymsl2QD^t?VnYmt%eYc=gJ0Y_Pb25>)EmL?$BwLg9gi`fk
zupIi8O>;y6Kls%UG(w{mf-tQ@uzF(5%H*@MM+h7Zw;Ph%##Hz!?zXq!Z{ZVbGEO_C
zz=&y{NleZzC8QaxkP+T9vi7tkrN#45#<I-3IFz2rOvye9$Wx5t4A0>t&pewILL^L6
zh#zSue}<sIQ8=RN+L7vzs;ezfqlQ!;U{8nN@Ie*SxNkL|2S{S6C8keNI_?A-u%orU
zVVV}QlW<G-jL8{}emu}p<rhMFBFEfx>$2J(^~tN8EEwYqlaXLdFg%YQ*rV6lQ}|cg
zn4Y917QzvHwSA@a-~v1yFQ;54#W^fXt>qUcgct&xZEkbp9b^Ca&Tz_s??9F%IvHua
zK3tFYJN>YN>DI^G;Rge%k0oUT=HhLbdT0-3Fx(Y^xABRt_22@9qUu4je72Grom)r?
z@de)N8ihsN8EnEABbD$aUW!b@SMGk~Lybx#qYuN^?mN*a8`|%-A&tGkx=J#f%Pt9_
z;|nM6Q6wJk#$;?D@ETj_WRxA=iTNQ+IT81d*v)c@ZrtgaD3=Kbgs|m;WzH-L5u+a|
zKR`$6v*LHLW_<RM-@VEcf$Lnbli{i3C-ESRx%=^p04dz-Kk1J2-354^%pNne{K8zp
z9es2hVAtL29|ic#ONl&w8{ixKe&841GF6J!;yC@UKb*GUE<QO}fIsoh;OEvXi_9@{
zjCls%om^t9Fjg4_#v0=q;|8P1SZBP!*kEijwg@GEPG)nKv{&R!EDfE7r?F%B6ugNS
HhA;mOP?KOJ

delta 992
zcmZ9J%TH556vlf>X({D`&=;kpElLScs<b@C3W6Gpk3@}PQWJ+lFOTwQnO*|L@=#Et
zAxLr-x-pc6F(d{tHzp(|qVcsdxHDm)B<j+Y|9~@v2I5Wb@0^)A=ljmRb8libE?gCa
zYz2Fkezy0VdMTVSs!y_2EP&hACisLWOlEv#EX596URyGES&>i^m6Vsrm<5&{NGQVj
zEatOVz+wT5`siu`!g{u1Z&*)1NQQ(W0kcZBW5i~Y#bP2lDKn{-G|M1mg|%T5$>A|+
z!gDr@P7zk6eGrNb<!RYPh&JyRUIf}9nfFQ86~D)a=1VOr>VP`d7mSMD-RKKwhJxc{
zP#oUM@1l`ls5=;w#1Kgp1yrH_A#aL4s%Kx0wnD4&mjWB@<wy$cIAzblg~H3Yrpv_b
z!q&1aU27=R7VI7v3)0YvhPqhKR%S%%6DjRvIVB7MY`564PRPT!MZk!o2(s~kqgwuE
zaRLPKyR%Bw#mX?LGvaZp!Pd<ioeV3glpq!PhL58Ly@DQRtXi-mDXEv=THBOG^aelT
z<L<z~vrYy-HrTH)hk5(5l7@JzRzpZ^w2P1idB8!_?)|Jb?XC;+ZYM|Yu9-@mY8jPD
zNvo2{Eg#wh5l2d0uz-(B=Y%LLu2Aldn(o0CS0=3EY1eu62p4Nf+YgJqguF)C)eUcC
zbJ?>DrnOPwfp_wsiV8l_=HW_gb)SVD`z3c4DA3_?;TQL3c6z_M5;v+Eb#X50<k%ZC
z8W|(RAUixSc68=<H6llHpj=KcrUbfliT|3lUI8x3X74qC9y#L6Rl*FL=^!R+RS=h-
z)^-9s!HCZ-d+SR8)>7$I!z#c>^c?*SG5lKZKwpyxKk#wWB>Y9wvGwU5Zqmmwz%j_d
ze-))162~aVIL8FXO^!*9DUNB5TO7AJW;x~<wz8r$5*4Wwr+4_?-p!}7)b$$`c)?!@
OEjaG4fxGzJfBhfIwhQn8

diff --git a/norch/tensor.py b/norch/tensor.py
index a4a84f6..4a896b3 100644
--- a/norch/tensor.py
+++ b/norch/tensor.py
@@ -223,6 +223,8 @@ class Tensor:
     def __add__(self, other):
         if isinstance(other, (int, float)):
             other = other * self.ones_like()
+        
+        broadcasted_shape = []
 
         # Function to determine if broadcasting is needed and get the broadcasted shape
         def broadcast_shape(shape1, shape2):
@@ -233,7 +235,7 @@ class Tensor:
             shape1 = [1] * (max_len - len(shape1)) + shape1
             shape2 = [1] * (max_len - len(shape2)) + shape2
 
-            broadcasted_shape = []
+            
             for dim1, dim2 in zip(shape1, shape2):
                 if dim1 != dim2 and dim1 != 1 and dim2 != 1:
                     raise ValueError("Shapes are not compatible for broadcasting")
@@ -255,7 +257,9 @@ class Tensor:
             result_data.ndim = len(broadcasted_shape)
 
             result_data.device = self.device
-            result_data.numel = self.numel  # Update this to calculate the correct number of elements if broadcasting
+            result_data.numel = 1
+            for s in result_data.shape:
+                result_data.numel *= s
 
             result_data.requires_grad = self.requires_grad or other.requires_grad
             if result_data.requires_grad: