From GITHUB 168

COPYBARA_INTEGRATE_REVIEW=https://api.github.com/repos/MegEngine/MegEngine/pulls/168 from Tricster <mediosrity@gmail.com> 5c775d02dd0b8f20b5acc6b400cf722e92f2e86b
Closes #168
GitOrigin-RevId: bceecb401b4bf5f6cfd47bb7f7216ec36ef1cae8