Olá, Prom = Baile Até mais!

Acredito que "baile de formatura" seja mais apropriado, apenas "baile" abrange muita coisa.

Just to add: Baile can refer to any kind of semi-formal or even casual dance gathering. If you want to refer specifically to the Prom, you can say "baile de formatura".