putting computation of gelman diagnostics on gpu