Diallo competed for the under-18 United States national basketball team that captured gold in the FIBA Americas Under-18 Championship game in 2016. He won bronze at the 2017 FIBA Under-19 Basketball World Cup in Egypt.