<gh_stars>10-100
import cupy as np


def kernel(alpha, beta, C, A, B):

    with np.cuda.Device(0):
        C[:] = alpha * A @ B + beta * C

    # with cp.cuda.Device(0):

    #     g_A = cp.asarray(A)
    #     g_B = cp.asarray(B)
    #     g_C = cp.asarray(C)

    #     g_C[:] = g_alpha * g_A @ g_B + g_beta * g_C
