MCPcopy Create free account
hub / github.com/pytorch/executorch / create_inspector

Function create_inspector

examples/apple/coreml/scripts/inspector_utils.py:194–238  ·  view source on GitHub ↗
(
    edge_program: EdgeProgramManager,
    executorch_program: ExecutorchProgramManager,
    example_inputs: Tuple[Value, ...],
    model_name: str,
    root_dir_path: Path,
    working_dir_path: Path,
    method_name: str = "forward",
    debug_buffer_size: int = 1 * 1024 * 1024 * 1024,
    delegate_metadata_parser=None,
    delegate_time_scale_converter=None,
)

Source from the content-addressed store, hash-verified

192
193
194def create_inspector(
195 edge_program: EdgeProgramManager,
196 executorch_program: ExecutorchProgramManager,
197 example_inputs: Tuple[Value, ...],
198 model_name: str,
199 root_dir_path: Path,
200 working_dir_path: Path,
201 method_name: str = "forward",
202 debug_buffer_size: int = 1 * 1024 * 1024 * 1024,
203 delegate_metadata_parser=None,
204 delegate_time_scale_converter=None,
205) -> Inspector:
206 et_record_path = working_dir_path / f"{model_name}_etrecord.bin"
207 generate_etrecord(
208 et_record=et_record_path.resolve(),
209 edge_dialect_program=edge_program,
210 executorch_program=executorch_program,
211 )
212
213 bundled_program_path = working_dir_path / f"{model_name}.bpte"
214 generate_bundled_program(
215 executorch_program=executorch_program,
216 example_inputs=example_inputs,
217 method_name=method_name,
218 bundled_program_path=bundled_program_path,
219 )
220
221 et_dump_path: Path = working_dir_path / f"{model_name}_etdump.etdp"
222 debug_buffer_path: Path = working_dir_path / f"{model_name}_debug_output.bin"
223 generate_etdump_with_intermediate_values(
224 root_dir_path=root_dir_path,
225 bundled_program_path=bundled_program_path,
226 et_dump_path=et_dump_path,
227 debug_buffer_path=debug_buffer_path,
228 debug_buffer_size=debug_buffer_size,
229 )
230
231 return Inspector(
232 etdump_path=str(et_dump_path.resolve()),
233 etrecord=str(et_record_path.resolve()),
234 debug_buffer_path=str(debug_buffer_path.resolve()),
235 enable_module_hierarchy=True,
236 delegate_metadata_parser=delegate_metadata_parser,
237 delegate_time_scale_converter=delegate_time_scale_converter,
238 )
239
240
241def parse_coreml_delegate_metadata(delegate_metadatas: List[str]) -> Dict[str, Any]:

Callers 2

create_inspector_coremlFunction · 0.85

Calls 4

generate_etrecordFunction · 0.90
InspectorClass · 0.90
generate_bundled_programFunction · 0.85

Tested by

no test coverage detected