Zwischenzeitlich zum General der Infanterie befördert, war Wartenberg im Oktober 1916 in die Schlacht an der Somme und im April 1917 in die Schlacht an der Aisne eingebunden. Er wurde mit dem Kronen-Orden I. Klasse mit Schwertern ausgezeichnet, verlegte mit seiner Division an die Ostfront und nahm hier an den Kämpfen um Riga teil. Nach fünf Monaten kehrte Wartenberg im September 1917 an die Westfront zurück, wo er am 5. November 1917 vor Verdun gefallen ist.