code: def _dgl_get_stream(ctx):
      """Get the current CUDA stream of the given DGL context.
  
      Parameters
      ----------
      ctx : DGL context.
  
      Returns
      -------
      DGLStreamHandle
          DGLStreamHandle of the current CUDA stream.
      """
      current_cuda_stream = DGLStreamHandle()
      check_call(
          _LIB.DGLGetStream(
              ctx.device_type,
              ctx.device_id,
              ctypes.byref(current_cuda_stream),
          )
      )
      return current_cuda_stream
  
imports: from __future__ import absolute_import
  import ctypes
  from .base import _FFI_MODE, _LIB, check_call
  from .runtime_ctypes import DGLStreamHandle
label_desc: Use image node_id 2 for calling a global function with example usage: _dgl_get_stream(ctx) and returns: current_cuda_stream