Skip to content

Commit fa9f6c5

Browse files
committed
Fixed bug causing undef var error when unrolling.
1 parent 6b717f8 commit fa9f6c5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/LoopVectorization.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -250,7 +250,7 @@ end
250250
end
251251
if !(N isa Integer) || r > 0
252252
masksym = gensym(:mask)
253-
masked_loop_body = add_masks(main_body, masksym, reduction_symbols, mod)
253+
masked_loop_body = add_masks(unrolled_loop_body_iter, masksym, reduction_symbols, mod)
254254
if N isa Integer
255255
push!(q.args, quote
256256
$masksym = $(VectorizationBase.mask(T, r))

0 commit comments

Comments
 (0)