add gpuId. The output data format is changed.

Signed-off-by: Enliang <ezheng@cs.unc.edu>