W reprezentacji Burkiny Faso Diarra zadebiutował w 1995 roku. W 1996 roku został powołany do kadry na Puchar Narodów Afryki 1996. Zagrał na nim w trzech meczach: ze Sierra Leone (1:2), z Zambią (1:5) i z Algierią (1:2).