now let's try to make it work on GPU