Remove comments in the GPU version
Some documentation updates