| import os.path as _osp |
| import sys |
| |
| from .throughput_benchmark import ThroughputBenchmark |
| from ._crash_handler import enable_minidumps, disable_minidumps, enable_minidumps_on_exceptions |
| |
| # Set the module for a given object for nicer printing |
| def set_module(obj, mod): |
| if not isinstance(mod, str): |
| raise TypeError("The mod argument should be a string") |
| obj.__module__ = mod |
| |
| if sys.executable == "torch_deploy": |
| # not valid inside torch_deploy interpreter, no paths exists for frozen modules |
| cmake_prefix_path = None |
| else: |
| cmake_prefix_path = _osp.join(_osp.dirname(_osp.dirname(__file__)), 'share', 'cmake') |
