def cublasSgemmEx()

in openai_gemm.py [0:0]

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

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