change the name to GpuGroupDotGradStep