Returning east, Griffin commanded the Second Brigade, Second Division, in the Battle of the Wilderness, Battle of Spotsylvania Court House and the Battle of Cold Harbor. He was appointed brigadier general of volunteers on May 30, 1864 to rank from May 12, 1864.