TorchFX caching fix (#18813)

* TorchFX caching fix

* Added required newlines for code formatting

* TorchFX model caching file_name check added

src/bindings/python/src/openvino/frontend/pytorch/torchdynamo/backend.py

@@ -127,3 +127,7 @@ def fx_openvino(subgraph, example_inputs):
     except Exception as e:
         log.debug(f"Failed in OpenVINO execution: {e}")
         return compile_fx(subgraph, example_inputs)
+
+
+def reset():
+    clear_caches()

src/bindings/python/src/openvino/frontend/pytorch/torchdynamo/compile.py

@@ -52,6 +52,8 @@ def openvino_compile(gm: GraphModule, *args, model_hash_str: str = None):
         for idx, input_data in enumerate(args):  # subgraph.example_inputs):
             input_types.append(input_data.type())
             input_shapes.append(input_data.size())
+            if file_name is not None:
+                file_name += "_" + str(input_data.type()) + str(input_data.size())[11:-1].replace(" ", "")
 
         decoder = TorchFXPythonDecoder(gm, gm, input_shapes=input_shapes, input_types=input_types)
 

src/bindings/python/src/openvino/frontend/pytorch/torchdynamo/execute.py

@@ -143,9 +143,20 @@ def openvino_execute_partitioned(gm: GraphModule, *args, executor_parameters=Non
 
     signature = str(id(gm))
     for idx, input_data in enumerate(args):
-        signature = signature + "_" + str(idx) + ":" + str(input_data.type())[6:] + ":" + str(input_data.size())[11:-1].replace(" ", "")
+        if isinstance(input_data, torch.Tensor):
+            signature = signature + "_" + str(idx) + ":" + str(input_data.type())[6:] + ":" + str(input_data.size())[11:-1].replace(" ", "")
+        else:
+            signature = signature + "_" + str(idx) + ":" + type(input_data).__name__ + ":val(" + str(input_data) + ")"
 
     if signature not in partitioned_modules:
         partitioned_modules[signature] = partition_graph(gm, use_python_fusion_cache=use_python_fusion_cache,
                                                          model_hash_str=model_hash_str)
     return partitioned_modules[signature](*args)
+
+
+def clear_caches():
+    global partitioned_modules
+    global compiled_cache
+
+    compiled_cache.clear()
+    partitioned_modules.clear()