fix up gpu test
fix up gpu test

GoogleCloudPlatform/testgrid,GoogleCloudPlatform/testgrid,GoogleCloudPlatform/testgrid,GoogleCloudPlatform/testgrid,GoogleCloudPlatform/testgrid,GoogleCloudPlatform/testgrid