Fix bug where grid.z is larger than 65535, there is a launch failure.

Fix bug where grid.z is larger than 65535, there is a launch failure. #126