CUDA不支持边界检查
Posted
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了CUDA不支持边界检查相关的知识,希望对你有一定的参考价值。
我正在尝试使用Numba并访问GPU以加速代码,但出现以下错误:
in jit raise NotImplementedError("bounds checking is not supported for CUDA")
NotImplementedError: bounds checking is not supported for CUDA
[我看到有人提出了另一个问题,但没有完全指明或回答here。当我看到矢量化代码(y = corr*x + np.sqrt(1.-corr**2)*z
)不起作用(相同错误)时,我实现了2-for循环。我也尝试使用选项boundscheck
,但这并没有改变结果。未指定target
时未出现错误,因为它会自动出现在CPU上(我想)。
import numpy as np
from numba import jit
N = int(1e8)
@jit(nopython=True, target='cuda', boundscheck=False)
def Brownian_motions(T, N, corr):
x = np.random.normal(0, 1, size=(T,N))
z = np.random.normal(0, 1, size=(T,N))
y = np.zeros(shape=(T,N))
for i in range(T):
for j in range(N):
y[i,j] = corr*x[i,j] + np.sqrt(1.-corr**2)*z[i,j]
return(x,y)
x, y = Brownian_motions(T = 500, N = N, corr = -0.45)
您能帮我吗? Python是3.7.6,Numba是0.48.0。
答案
只需解决,您可以通过从@
中删除@jit
来做到这一点>以上是关于CUDA不支持边界检查的主要内容,如果未能解决你的问题,请参考以下文章
RuntimeError: ‘lengths’ argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor(代码片段