Move the definiton of GpuGetContext into context.py.