pt: remove env.DEVICE in all forward
functions
#3330
Merged
Loading
forward
functions
#3330