Efficient non-uniform grid for GPU-parallel shallow water equations models