Exemplos de cublasXtSetBlockDim em Python

Exemplo n.º 1

def cublasXtGetBlockDim(handle):
    blockDim = ctypes.c_void_p()
    status = _libcublas.cublasXtSetBlockDim(handle, ctypes.byref(blockDim))
    cublasCheckStatus(status)
    return blockDim.value

Exemplo n.º 2

def cublasXtSetBlockDim(handle, blockDim):
    status = _libcublas.cublasXtSetBlockDim(handle, blockDim)
    cublasCheckStatus(status)