Code

[PT FE] Add ModuleExtension#23536

Code

Merged

mvafin merged 23 commits intoopenvinotoolkit:masteropenvinotoolkit/openvino:masterfrom mvafin:mvafin/pt_fe/module_extensionmvafin/openvino:mvafin/pt_fe/module_extensionCopy head branch name to clipboardMar 20, 2024

+231-29

Commits

Commits on Feb 2, 2024

• Draft of ModuleExtension class which triggers PyTorch model patching with module replacemnts by PythonOp nodes

  slyalin

  committed
  3f1bc39

  Copy full SHA for 3f1bc39

Commits on Feb 5, 2024

• Provided a stub op PagedAttentionPlaceholder that translates input shape to output and doesn't have evaluate implementation

  slyalin

  committed
  bbd11ec

  Copy full SHA for bbd11ec

• Merge remote-tracking branch 'origin/master' into pytorch_module_extension

  slyalin

  committed
  6d9c2fb

  Copy full SHA for 6d9c2fb

• Wrapper to glue custom op and tracable PyTorch code

  slyalin

  committed
  4a2f534

  Copy full SHA for 4a2f534

Commits on Feb 6, 2024

• Added a stub for PageAttentionPlaceholder to run in CPU plugin

  slyalin

  committed
  2d4a145

  Copy full SHA for 2d4a145

Commits on Feb 7, 2024

• Cleanup, more checks, removed PagedAttentionPlacehoder, more accurate extension handling

  slyalin

  committed
  e1cb833

  Copy full SHA for e1cb833

Commits on Feb 8, 2024

• Fixed model unpatching when using ModuleExtension

  slyalin

  committed
  86e0263

  Copy full SHA for 86e0263

• Fixed missing sharing of torch.Tensor when passing to OV

  slyalin

  committed
  2af4abc

  Copy full SHA for 2af4abc

Commits on Feb 12, 2024

• Fix relative paths for shared library extensions in convert_model

  slyalin

  committed
  a157c65

  Copy full SHA for a157c65

• Merged with master

  slyalin

  committed
  b4293f1

  Copy full SHA for b4293f1

• Merged from master

  slyalin

  committed
  5c6218e

  Copy full SHA for 5c6218e

Commits on Feb 13, 2024

• Added support for OPENVINO_FRAMEWORK_MAP(pytorch), including convert_model.

  slyalin

  committed
  ac97fa8

  Copy full SHA for ac97fa8

• Merge remote-tracking branch 'origin/master' into pytorch_module_extension

  slyalin

  committed
  0ead7f1

  Copy full SHA for 0ead7f1

Commits on Feb 15, 2024

• Merged from master

  slyalin

  committed
  95b9f9e

  Copy full SHA for 95b9f9e

• ModuleExtension: documentation, new names for arguments, reasonable default values for evaluate/convert members.

  slyalin

  committed
  c81019c

  Copy full SHA for c81019c

Commits on Feb 26, 2024

• Merge branch 'master' into pytorch_module_extension

  slyalin

  authored
  757caa1

  Copy full SHA for 757caa1

Commits on Mar 11, 2024

• Add tests

  mvafin

  committed
  a16b402

  Copy full SHA for a16b402

Commits on Mar 19, 2024

• Merge remote-tracking branch 'upstream/master' into mvafin/pt_fe/module_extension

  mvafin

  committed
  a9ce494

  Copy full SHA for a9ce494

• Fix style

  mvafin

  committed
  e8b8c55

  Copy full SHA for e8b8c55

• Merge branch 'master' into mvafin/pt_fe/module_extension

  mvafin

  authored
  ebfe4c1

  Copy full SHA for ebfe4c1

• Remove debug print

  mvafin

  committed
  9f4813e

  Copy full SHA for 9f4813e

• Update tools/ovc/openvino/tools/ovc/moc_frontend/pytorch_frontend_utils.py

  mvafin

  authored
  1867c33

  Copy full SHA for 1867c33

Commits on Mar 20, 2024

• Merge branch 'master' into mvafin/pt_fe/module_extension

  mvafin

  authored
  25f9eb2

  Copy full SHA for 25f9eb2