[ats] Export more functions

The MAIN_MODULE fix wasn't tested correctly and it turns out we still
need to manually export more symbols.