Mit der Schweizer Eishockeynationalmannschaft nahm Diaz an der U18-Junioren-B-Weltmeisterschaft 2004 sowie den U20-Junioren-Weltmeisterschaften 2005 und 2006 teil. Des Weiteren stand er im Aufgebot der Schweiz bei den Weltmeisterschaften 2008 und 2011 und nahm an den Olympischen Winterspielen 2010 in Vancouver teil.