IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[PT FE]: switch on tracing as main path if example inputs provided (#17194)

Browse Source
This commit is contained in:
Ekaterina Aidova 2023-04-26 12:50:43 +04:00 committed by GitHub
parent 09265083ed
commit 5857c4438b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/bindings/python/src/openvino/frontend/pytorch/decoder.py
View File

@ -141,17 +141,17 @@ class TorchScriptPythonDecoder (Decoder):
input_signature = None input_signature = None
if isinstance(pt_module, torch.nn.Module) and not isinstance(pt_module, (torch.jit._trace.TopLevelTracedModule, torch.jit._script.RecursiveScriptModule)): if isinstance(pt_module, torch.nn.Module) and not isinstance(pt_module, (torch.jit._trace.TopLevelTracedModule, torch.jit._script.RecursiveScriptModule)):
input_signature = list(inspect.signature(pt_module.forward).parameters.keys()) input_signature = list(inspect.signature(pt_module.forward).parameters.keys())
try: if example_inputs is None:
scripted = torch.jit.script(pt_module) scripted = torch.jit.script(pt_module)
except Exception as scripting_err: else:
if example_inputs is not None: inputs, input_signature = prepare_example_inputs(example_inputs, input_signature)
inputs, input_signature = prepare_example_inputs(example_inputs, input_signature) try:
scripted = torch.jit.trace(pt_module, inputs)
except Exception:
try: try:
scripted = torch.jit.trace(pt_module, inputs) scripted = torch.jit.script(pt_module)
except Exception as tracing_e: except Exception:
raise tracing_e scripted = torch.jit.trace(pt_module, inputs, strict=False)
else:
raise scripting_err
else: else:
scripted = pt_module scripted = pt_module
if freeze: if freeze:
@ -253,7 +253,7 @@ class TorchScriptPythonDecoder (Decoder):
def get_subgraph_size(self) -> int: def get_subgraph_size(self) -> int:
if isinstance(self.graph_element, torch.Node): if isinstance(self.graph_element, torch.Node):
return len(self.get_subgraphs()) return len(self.get_subgraphs())
else: else:
return 1 return 1