Fix some compilation crash of the new GpuSubtensor
Make the new GpuSubtensor tests run with it!