Как исправить ошибку "typingerror: невозможно определить тип numba для <class 'function'>?

Я использую Numba для работы с массивом NumPy. Когда я переключаю цель в @jit(target = ___) с 'cpu' на 'cuda' или 'parallel', я получаю ошибку в заголовке.

Когда я гуглю эту ошибку, решения, которые я вижу, говорят, что программист должен использовать массивы numpy, что я и делаю. ( numba - TypingError: невозможно определить тип Numba для )

import numpy as np
import multiprocessing
from multiprocessing import Process, freeze_support  
from numba import vectorize, cuda, float64, float32, int64
from numba.cuda import random
from numba import vectorize, cuda, jit, njit, prange
from numba.numpy_support import from_dtype
import numba

array = np.zeros((3, 5), dtype = np.float64)
array[:,0] = 2
length = (len(array[0]))
second_array = np.empty_like(array)

@jit(target = 'cpu')

def func(list):
    for i in range(length-1):   
        variable = (np.random.poisson( 15, 1))       
        list[i+1] = list[i] + variable
    return(list)
    print(list)

@jit(target = 'cpu')
#I want the target to be 'cuda'

def func_two(input_array):
    for i in range(len(input_array)):
        second_array[i] = func(input_array[i])
    return(second_array)

(func_two(array)

0 ответов

Другие вопросы по тегам