def cublasHgemm()

in openai_gemm.py [0:0]

• 12 lines of code
• 1 McCabe index (conditional complexity)

def cublasHgemm(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta, C, ldc):

    alpha = unpack_from('H', h_dtype.type(alpha))[0]
    beta  = unpack_from('H', h_dtype.type(beta))[0]

    status = cublas._libcublas.cublasHgemm(handle,
        cublas._CUBLAS_OP[transa], cublas._CUBLAS_OP[transb],
        m, n, k,
        ctypes.byref(ctypes.c_uint16(alpha)),
        int(A), lda,
        int(B), ldb,
        ctypes.byref(ctypes.c_uint16(beta)),
        int(C), ldc)
    cublas.cublasCheckStatus(status)