Function EnsureCuBlasWorkspace

ngscuda/cuda_linalg.cpp:37–42

Call before graph capture to ensure workspace is allocated

  // Call before graph capture to ensure workspace is allocated
  void EnsureCuBlasWorkspace()
  {
    if (!cublas_workspace)
      cudaMalloc(&cublas_workspace, cublas_workspace_size);
    cublasSetWorkspace(Get_CuBlas_Handle(), cublas_workspace, cublas_workspace_size);
  }

  // Wrapper: set stream AND re-apply workspace if already allocated
  // (cublasSetStream resets workspace to default pool)