pytorch
bc0e216d - [jit][edge] Print correct type strings in code file for mobile models. (#71968)

Commit
3 years ago
[jit][edge] Print correct type strings in code file for mobile models. (#71968) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/71968 Right now when we output type to python files under `code/`, we directly write the dynamic type representation `Dynamic<>`, which causes server side to load an unsupported type. Instead we should do the fallback in export_module.cpp. ghstack-source-id: 147856473 Test Plan: CI buck test //xplat/pytorch/mobile/test:test_read_all_mobile_model_configs ``` ... [ OK ] GeneralAndSpecial/BackPortTest.BackPortForChunkIdx/37 (39142 ms) [ RUN ] GeneralAndSpecial/BackPortTest.BackPortForChunkIdx/38 total: 6 success: 6 failure: 0 [ OK ] GeneralAndSpecial/BackPortTest.BackPortForChunkIdx/38 (9651 ms) [ RUN ] GeneralAndSpecial/BackPortTest.BackPortForChunkIdx/39 total: 4 success: 4 failure: 0 [ OK ] GeneralAndSpecial/BackPortTest.BackPortForChunkIdx/39 (5509 ms) [----------] 40 tests from GeneralAndSpecial/BackPortTest (806244 ms total) [----------] Global test environment tear-down [==========] 41 tests from 2 test cases ran. (810453 ms total) [ PASSED ] 41 tests. ``` Reviewed By: pavithranrao Differential Revision: D33830355 fbshipit-source-id: 0be608fadf14daa2b703f31118ab648cb7b75f9b (cherry picked from commit 6d65049ae5ac1ef6a11d19de48dd4d926b793b34)
Author
Committer
Parents
Loading