Bug fix: GPU-timing syncronization

A problem was observed with different NVTX ranges on CPU and GPU. The problem has been solved by explicitly synchronizing the CPU with the GPU.