标签：__ python print k0 device cuda 计算 np GPU

A=cuda.to_device(a)
fgriddim, blockdim
b=A.copy_to_host()

cuda{
blockDim:[x],
blockIdx:[x],

threadDim:[x],
threadIdx:[x],

gridDim:[x],

}

from numba import cuda
import numpy as np

@cuda.jit
def addGPU(A,B,C):
    i=cuda.blockIdx.x
    j=cuda.threadIdx.x
    C[i][j]=A[i][j]+B[i][j]
@cuda.jit
def mulGPU(A,B,C,k):
    i=cuda.blockIdx.x
    j=cuda.threadIdx.x
    C[i][j]=0
    for k0 in range(k):
        C[i][j]=C[i][j]+A[i][k0]*B[k0][j]

if __name__ == '__main__':
    
    print("start")
    N=3
    A=np.array([
        [1,0,0],
        [0,2,0],
        [0,0,1]
    ])
    B=np.array([
        [1,0,0],
        [0,1,1],
        [0,1,1]
    ])
    d_A = cuda.to_device(A)
    d_B = cuda.to_device(B)
    d_C = cuda.to_device(np.zeros((N,N)))
    print("A:\n",A,"\nB:\n",B)
    addGPU[N, N](d_A,d_B,d_C)
    C = d_C.copy_to_host()
    print("add:\n",C)
    mulGPU[N, N](d_A,d_B,d_C,3)
    C = d_C.copy_to_host()
    print("mul:\n",C)

参考：https://blog.csdn.net/sazass/article/details/108649728

标签：__,python,print,k0,device,cuda,计算,np,GPU
来源： https://blog.csdn.net/liuzhicheng1845/article/details/122523317

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。