pytorch
747f27a9 - [Mobile] Update build_mobile.sh to allow lite interpreter and tracing based builds (#84647)

Commit
2 years ago
[Mobile] Update build_mobile.sh to allow lite interpreter and tracing based builds (#84647) Summary: Currently, build_mobile.sh doesn't allow lite interpreter builds or tracing based selective builds. build_mobile.sh is used for host builds of PyTorch for Mobile deployment. Additionally, certain flags such as `USE_BLAS` were not being respected as they should be. This change addresses that as well. Test Plan: Build using: ``` cat /tmp/selected_ops.yaml - aten::add - aten::sub ``` ``` BUILD_PYTORCH_MOBILE_WITH_HOST_TOOLCHAIN=1 USE_LIGHTWEIGHT_DISPATCH=0 BUILD_LITE_INTERPRETER=1 SELECTED_OP_LIST=/tmp/selected_ops.yaml ./scripts/build_mobile.sh ``` ``` cat /tmp/main.cpp int main() { auto m = torch::jit::_load_for_mobile("/tmp/path_to_model.ptl"); auto res = m.forward({}); return 0; } ``` Test using: ``` g++ /tmp/main.cpp -L build_mobile/lib/ -I build_mobile/install/include/ -lpthread -lc10 -ltorch_cpu -ltorch -lXNNPACK -lpytorch_qnnpack -lcpuinfo -lclog -lpthreadpool -lgloo -lkineto -lfmt -ldl -lc10 ``` Reviewers: Subscribers: Tasks: Tags: Pull Request resolved: https://github.com/pytorch/pytorch/pull/84647 Approved by: https://github.com/JacobSzwejbka, https://github.com/cccclai
Author
Committer
Parents
Loading