| 41 | int device_id() const { return m_device_id; } |
| 42 | |
| 43 | cudaStream_t stream() const { return megcore_context().stream; } |
| 44 | cudnnHandle_t cudnn_handle() { return m_cudnn_handle; } |
| 45 | cublasHandle_t cublas_handle() { return m_cublas_handle; } |
| 46 | #if CUDA_VERSION >= 10010 |