Skip to content
  • Sebastian Ohlmann's avatar
    Set stream for cuBlas functions · b890ed7e
    Sebastian Ohlmann 创作于
    This is needed to ensure the cublas functions are executed in the same
    stream as the rest of the CUDA kernels and thus are synchronous to the
    other operations.
    b890ed7e