Skip to content

Commit ae5d6b1

Browse files
committed
initialize registers
1 parent de767e3 commit ae5d6b1

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

src/memopt.jl

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -117,6 +117,16 @@ function memopt(metadata_module::Module, is_parallel_kernel::Bool, caller::Modul
117117

118118
body = quote
119119
$loopoffset = (@blockIdx().y-1)*$loopsize + $range_y_start-1
120+
$((:( $reg = 0.0
121+
)
122+
for A in optvars for regs in values(regqueue_tails[A]) for reg in values(regs)
123+
)...
124+
)
125+
$((:( $reg = 0.0
126+
)
127+
for A in optvars for regs in values(regqueue_heads[A]) for reg in values(regs)
128+
)...
129+
)
120130
$((wrap_loop(i, loopstart:0,
121131
quote
122132
$iy = $i + $loopoffset

0 commit comments

Comments
 (0)