Some simplification and better reproducability (tested on Ray with device cuda).
Some simplification and better reproducability (tested on Ray with device cuda).

mfem/mfem,mfem/mfem,mfem/mfem,mfem/mfem,mfem/mfem