ModuleExtension, a new type of extension for PyTorch frontend#22867

Closed

slyalin wants to merge 16 commits intoopenvinotoolkit:masterfrom slyalin:pytorch_module_extension

+175-5

Commits on Feb 2, 2024

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

Commits on Feb 5, 2024

Provided a stub op PagedAttentionPlaceholder that translates input shape to output and doesn't have evaluate implementation
slyalin
committed
Merge remote-tracking branch 'origin/master' into pytorch_module_extension
slyalin
committed
Wrapper to glue custom op and tracable PyTorch code
slyalin
committed

Commits on Feb 6, 2024

Added a stub for PageAttentionPlaceholder to run in CPU plugin
slyalin
committed

Commits on Feb 7, 2024

Cleanup, more checks, removed PagedAttentionPlacehoder, more accurate extension handling
slyalin
committed

Commits on Feb 8, 2024

Commits on Feb 12, 2024

Commits on Feb 13, 2024

Commits on Feb 15, 2024

Commits on Feb 26, 2024

Merge branch 'master' into pytorch_module_extension
slyalin
authored