Skip to content
This repository was archived by the owner on Feb 2, 2024. It is now read-only.

Commit 17cbf05

Browse files
author
Ehsan Totoni
committed
col filter put arrayset last for easier fusion
1 parent 9ff770d commit 17cbf05

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

hpat/hiframes.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -420,10 +420,12 @@ def fix_series_filter(self, blocks):
420420
index_var = stmt.value.index
421421
def f(A, B, ind):
422422
for i in numba.parfor.prange(len(A)):
423+
s = 0
423424
if ind[i]:
424-
A[i] = B[i]
425+
s = B[i]
425426
else:
426-
A[i] = np.nan
427+
s= np.nan
428+
A[i] = s
427429
f_blocks = get_inner_ir(f)
428430
replace_var_names(f_blocks, {'A': lhs.name})
429431
replace_var_names(f_blocks, {'B': in_arr.name})

0 commit comments

Comments
 (0)