Add test data shard for plain gpu version(tested) cpu version(not tested)