code: def _retain_primary_context(dev):
      context = dev.retain_primary_context()
      context.push()
      return context
  
imports: import pycuda.driver
  import atexit
  from pycuda.tools import make_default_context
label_desc: Use image node_id 1 for calling a global function with example usage: _retain_primary_context(dev) and returns: context