pytorch/tools/setup_helpers/gen_unboxing.py
Mengwei Liu 5889eae906 [retake][mobile] Fix lightweight dispatch OOM error by introducing selective build
See #78983 for more details. This PR adds a new option argument to avoid changing the existing one and added unit tests to cover the arugments.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/79215

Approved by: https://github.com/cccclai
2022-06-13 17:23:02 +00:00

12 lines
263 B
Python

# Little stub file to get BUILD.bazel to play along
import os.path
import sys
root = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
sys.path.insert(0, root)
import tools.jit.gen_unboxing
tools.jit.gen_unboxing.main(sys.argv[1:])