Made timeloop work on GPU again