(
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,
)
| 192 | |
| 193 | |
| 194 | def 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 | |
| 241 | def parse_coreml_delegate_metadata(delegate_metadatas: List[str]) -> Dict[str, Any]: |
no test coverage detected