ton/IronpenGPU
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add alif learn

Browse Source
This commit is contained in:
ton
2023-08-26 15:02:41 +07:00
parent ef72f7949a
commit eb4f22de11
3 changed files with 120 additions and 153 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/forward.jl
View File

@@ -243,7 +243,7 @@ function lifForward( zit,
linear_to_cartesian,
)
i = (blockIdx().x - 1) * blockDim().x + threadIdx().x # gpu threads index
if i <= length(wRec)
# cartesian index
i1, i2, i3, i4 = linear_to_cartesian(i, size(wRec))
@@ -416,7 +416,7 @@ function alifForward( zit,
linear_to_cartesian,
)
i = (blockIdx().x - 1) * blockDim().x + threadIdx().x # gpu threads index
if i <= length(wRec)
# cartesian index
i1, i2, i3, i4 = linear_to_cartesian(i, size(wRec))