Fix device error when using cuda

#4
by majinyu - opened
No description provided.
xinyu1205 changed pull request status to merged

Sign up or log in to comment