change device selection a little bit

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