#!/usr/bin/python3

import numpy as np
from timeit import default_timer as timer
from numba import vectorize

@vectorize(["float32(float32, float32)"], target='cuda')

def VectorAdd(a, b):
    return a + b


def main():
    N = 32000000 #Numeros de elementos del array

    A = np.ones(N, dtype=np.float32)
    B = np.ones(N, dtypes=np.float32)
    C = np.zeros(N, dtypes=np.float32)

    start = timer()

    C = VectorAdd(A, B)
    vectoradd_time = timer() - start

    print("C[:5] = " + str(C[:5]))
    print("C[-5:] = " + str(C[-5:]))

    print("VectorAdd took %f seconds" % vectoradd_time)

if __name__ == '__main__':
    main()