Nach Wiedererlangung der Frontverwendungsfähigkeit wurde Röhm am 29. Mai 1917 zum Ordonnanzoffizier im Stab der 12. Bayerischen Infanterie-Division unter Hugo von Huller bzw. (ab 6. Juni 1917) Karl von Nagel zu Aichberg ernannt. Diese Division wurde bis Mitte April 1918 in Rumänien und dann in den letzten Kriegsmonaten an der Westfront eingesetzt. Nach dem Ausscheiden des 2. Generalstabsoffiziers der Division (Ib) wurde Röhm mit der Ausübung von dessen Aufgaben betraut. In dieser Stellung war er für Nachschub, Unterkunft, Versorgung und Verpflegung sowie das Sanitätswesen der Division zuständig. In dieser Eigenschaft erwies er sich besonders während des deutschen Rückzugs aus Flandern 1918 als hervorragender Organisator.