Added basic functionality to run PTM on GPU (incomplete)