W reprezentacji Mali Diarra zadebiutował w 1997 roku. W 2002 roku został powołany do kadry na Puchar Narodów Afryki 2002. Na tym turnieju był podstawowym zawodnikiem Mali i wystąpił w 5 meczach: grupowych z Liberią (1:1), z Nigerią (0:0) i z Algierią (2:0), ćwierćfinałowym z Republiką Południowej Afryki (0:2) i półfinale z Kamerunem (0:3). Od 1997 do 2004 roku rozegrał w kadrze narodowej 22 mecze.