Я пытаюсь запустить следующее ядро
from numba import cuda
import numpy as np
@cuda.jit
def Dark_Pix_Search_Kernel(all_bands,window_size,output_array):
row_size = all_bands.shape[0]
col_size = all_bands.shape[1]
end_r = row_size - (row_size -row_size*int(row_size/window_size[0])) - window_size[0] + 1
end_c = col_size - (row_size -row_size*int(row_size/window_size[0])) - window_size[0] + 1
row,col,band_n = cuda.grid(3)
if (row in range(0,end_r,window_size[0])) and (col in range(0,end_c,window_size[0])):
row_e = row+window_size[0]
col_e = col+window_size[0]
output_array [row][col][band_n] = np.median(all_bands[row:row_e, col:col_e,band_n])
, но получаю следующую ошибку
LoweringError: No definition for lowering <built-in method impl_contains_helper of _dynfunc._Closure object at 0x000002851EB23F98>(range_state_int64, int32) -> bool