When Major General Robert B. Potter, the division commander, was wounded in an attack on Fort Mahone during the Third Battle of Petersburg on April 2, 1865, Griffin became acting division commander during the Appomattox Campaign. He also commanded the division in the Department of Washington after the Confederate surrender.