Как запустить эту функцию на моем GPU? (python) - PullRequest
Новая
       14

Как запустить эту функцию на моем GPU? (python)

2 голосов
/ 28 апреля 2020

Я бы хотел ускорить эту программу, используя мой gpu: 

import string
from numba.typed import List
from numba import jit, cuda
import time

alp = string.ascii_letters


@jit(target="gpu")
def test():
    for i in alp:
        for x in alp:
            print(i + x)

def speedTest():
    start_time = time.time()
    test()
    print(time.time() - start_time)

Но каждый раз, когда я получаю эту ошибку: 

Traceback (most recent call last):
  File "<pyshell#15>", line 1, in <module>
    speed()
  File "F:/Script Projects#/Tester.py", line 17, in speed
    test()
  File "D:\Python\lib\site-packages\numba\cuda\dispatcher.py", line 40, in __call__
    return self.compiled(*args, **kws)
  File "D:\Python\lib\site-packages\numba\cuda\compiler.py", line 758, in __call__
    kernel = self.specialize(*args)
  File "D:\Python\lib\site-packages\numba\cuda\compiler.py", line 769, in specialize
    kernel = self.compile(argtypes)
  File "D:\Python\lib\site-packages\numba\cuda\compiler.py", line 784, in compile
    kernel = compile_kernel(self.py_func, argtypes,
  File "D:\Python\lib\site-packages\numba\core\compiler_lock.py", line 32, in _acquire_compile_lock
    return func(*args, **kwargs)
TypeError: compile_kernel() got an unexpected keyword argument 'boundscheck'

Я хочу позже заменить 'print () «с чем-то еще, но сейчас мне просто нужно, чтобы это работало. Он работает без режима python вместо target = "gpu". Спасибо за любую помощь!

...