change the way to load data on gpu