Add some generic code for vGPU testing