Support is_mtia() (#106396)
Summary: As title.
Test Plan: CI tests.
Reviewed By: yuhc
Differential Revision: D47937061
Pull Request resolved: https://github.com/pytorch/pytorch/pull/106396
Approved by: https://github.com/yuhc, https://github.com/ezyang
