multiply-by-2-full using numba and cuda

multiply-by-2-full.py

import numpy as np
from numba import cuda

@cuda.jit
def multiply_by_2(array):
    pos = cuda.grid(1)
    array[pos] = array[pos] * 2

# criamos um vetor ordenado até 1000, [0, 1, 2]
an_array = np.arange(1000)

threadsperblock = 32
blockspergrid = (an_array.size + (threadsperblock - 1)) // threadsperblock
multiply_by_2[blockspergrid, threadsperblock](an_array)

print(an_array)