code: def gpu_mem_trace(func):
      "A decorator that runs `GPUMemTrace` w/ report on func"
      @functools.wraps(func)
      def wrapper(*args, **kwargs):
          with GPUMemTrace(ctx=func.__qualname__, on_exit_report=True):
              return func(*args, **kwargs)
      return wrappef gpu_mem_trace(func):
      "A decorator that runs `GPUMemTrace` w/ report on func"
      @functools.wraps(func)
      def wrapper(*args, **kwargs):
          with GPUMemTrace(ctx=func.__qualname__, on_exit_report=True):
              return func(*args, **kwargs)
      return wrapper
docs: A decorator that runs `GPUMemTrace` w/ report on func