头歌Python 使用GPU加速计算

第一关

def sum_cpu(arr):
    '''设计一个二维累加运算，求数组所有元素的和'''
    # ********* Begin ******** #
    sum=0;
    for i in range(0,3):
        for j in range(0,3):
            sum+=arr[i,j]
    # ********** End ********* #


'''添加装饰器'''
# ******** Begin ******** #
@jit
def jitsum(arr):
    sum=0;
    for i in range(0,3):
        for j in range(0,3):
            sum+=arr[i,j]
# ******** End ********* #
def sum_jit(arr):
    '''设计一个二维累加运算，求数组所有元素的和'''
    # ********* Begin ******** #
    jitsum(arr)
    # ********** End ********* #

第二关

# *********** Begin *************#
@vectorize
def vector_add_gpu(a,b):
    return a+b

# ************* End *************#