Mit seiner Armee unterstützte er nach dem Scheitern der Ardennenoffensive die während der Schlacht um Budapest durch die Rote Armee eingeschlossenen deutschen Truppen. Am Ende des Krieges wurde Dietrich von Anfang April 1945 an mit der 6. SS-Panzerarmee als militärischer „Stadtkommandant“ der Schlacht um Wien tätig.