Code

Torch Compile - New Op Support#23310

Code

Merged

mvafin merged 78 commits intoopenvinotoolkit:masteropenvinotoolkit/openvino:masterfrom cavusmustafa:torch_compile/new_op_support_3cavusmustafa/openvino:torch_compile/new_op_support_3Copy head branch name to clipboardMar 21, 2024

+759-75

Commits

Commits on Mar 7, 2024

• New ops added to torch.compile support list

  cavusmustafa

  committed
  18212e1

  Copy full SHA for 18212e1

• Additional ops for NetVLad and ALIKE

  cavusmustafa

  committed
  b8c3575

  Copy full SHA for b8c3575

• Additional op support for ChatGLM2

  cavusmustafa

  committed
  73b25f7

  Copy full SHA for 73b25f7

• PTFE input_model constructor output access fix

  cavusmustafa

  committed
  1d1093b

  Copy full SHA for 1d1093b

• Removed bitwise_not fx version

  cavusmustafa

  committed
  25990fd

  Copy full SHA for 25990fd

• Additional op support for TorchFX

  cavusmustafa

  committed
  4b1f4f1

  Copy full SHA for 4b1f4f1

• Stack translation fix for TorchFX

  cavusmustafa

  committed
  eb8c2a0

  Copy full SHA for eb8c2a0

• TorchFX unit tests for Div and Elu

  cavusmustafa

  committed
  d9f792d

  Copy full SHA for d9f792d

• TorchFX unit test update: full, comparison, glu

  cavusmustafa

  committed
  8365227

  Copy full SHA for 8365227

• TorchFX unit tests: grip_sample_2d, hardswish, hardtanh

  cavusmustafa

  committed
  f4ec62e

  Copy full SHA for f4ec62e

• TorchFX unit tests: index_select, unary_ops, isinf, isnan

  cavusmustafa

  committed
  27f8e66

  Copy full SHA for 27f8e66

• TorchFX: Additional op unit tests

  cavusmustafa

  committed
  da180f6

  Copy full SHA for da180f6

• TorchFX: Additonal unit tests

  cavusmustafa

  committed
  d6d8121

  Copy full SHA for d6d8121

• Code style fix src/frontends/pytorch/src/op/cat.cpp

  Show description for 894eecf
  

  cavusmustafa

  and

  github-actions[bot]

  committed
  894eecf

  Copy full SHA for 894eecf

• Code style fix src/frontends/pytorch/src/op/any.cpp

  Show description for 2f2f7b9
  

  cavusmustafa

  and

  github-actions[bot]

  committed
  2f2f7b9

  Copy full SHA for 2f2f7b9

• Code style fix src/frontends/pytorch/src/op/any.cpp

  Show description for f1b546c
  

  cavusmustafa

  and

  github-actions[bot]

  committed
  f1b546c

  Copy full SHA for f1b546c

• Code style fix src/frontends/pytorch/src/op/isinf.cpp

  Show description for 04f6423
  

  cavusmustafa

  and

  github-actions[bot]

  committed
  04f6423

  Copy full SHA for 04f6423

• Code style fix src/frontends/pytorch/src/op/isinf.cpp

  Show description for 551f7d7
  

  cavusmustafa

  and

  github-actions[bot]

  committed
  551f7d7

  Copy full SHA for 551f7d7

• Code style fix src/frontends/pytorch/src/op/isnan.cpp

  Show description for 2d64021
  

  cavusmustafa

  and

  github-actions[bot]

  committed
  2d64021

  Copy full SHA for 2d64021

• Code style fix src/frontends/pytorch/src/op/isnan.cpp

  Show description for 56d7908
  

  cavusmustafa

  and

  github-actions[bot]

  committed
  56d7908

  Copy full SHA for 56d7908

• Code style fix src/frontends/pytorch/src/op/topk.cpp

  Show description for c1ab831
  

  cavusmustafa

  and

  github-actions[bot]

  committed
  c1ab831

  Copy full SHA for c1ab831

• Added embedding_bag and fixed unbind int

  

  ynimmaga

  authored and

  cavusmustafa

  committed
  cebbca1

  Copy full SHA for cebbca1

• Code style fix src/frontends/pytorch/src/op/split.cpp

  Show description for ebd8c80
  

  cavusmustafa

  and

  github-actions[bot]

  committed
  ebd8c80

  Copy full SHA for ebd8c80

• TorchFX: Unit test enabled for embedding_bag

  cavusmustafa

  committed
  207c25d

  Copy full SHA for 207c25d

• TorchFX: bitwise_right_shift is temporarily removed

  cavusmustafa

  committed
  19fc3c1

  Copy full SHA for 19fc3c1

• Removed unnecessary include lines from bitwise translation

  cavusmustafa

  committed
  816b2b8

  Copy full SHA for 816b2b8

• IsNan and IsInf translations are converted to 1to1

  cavusmustafa

  committed
  98ce772

  Copy full SHA for 98ce772

• Removed prim ops and unused torchvision ops

  cavusmustafa

  committed
  12bb8d9

  Copy full SHA for 12bb8d9

• import placement update for select_scatter unit test

  cavusmustafa

  committed
  c23809a

  Copy full SHA for c23809a

• Removed test_log.py file

  cavusmustafa

  committed
  53490df

  Copy full SHA for 53490df

Commits on Mar 8, 2024

• Bugfix src/frontends/pytorch/src/op/sort.cpp

  Show description for c7fbcb7
  

  cavusmustafa

  and

  mvafin

  authored
  c7fbcb7

  Copy full SHA for c7fbcb7

• var_correction_fx translation removed

  cavusmustafa

  committed
  1868d7e

  Copy full SHA for 1868d7e

• TorchFX: op list sorting

  cavusmustafa

  committed
  af3f54a

  Copy full SHA for af3f54a

• Revert change for tests/layer_tests/pytorch_tests/test_batch_norm.py

  cavusmustafa

  committed
  e912a2f

  Copy full SHA for e912a2f

• Revert the change for tests/layer_tests/pytorch_tests/test_flip.py

  cavusmustafa

  committed
  82a4389

  Copy full SHA for 82a4389

• Fixed mistakenly removed line in test_masked_fill.py

  cavusmustafa

  committed
  da637e5

  Copy full SHA for da637e5

• Typo fix in test_torch_decoder.py

  cavusmustafa

  committed
  e240e33

  Copy full SHA for e240e33

• Common sort translation for FX and TS

  cavusmustafa

  committed
  d188fba

  Copy full SHA for d188fba

• hardswish unit test moved into unary_ops

  cavusmustafa

  committed
  6f73d19

  Copy full SHA for 6f73d19

• new line fix in var_mean.cpp

  cavusmustafa

  committed
  5bcd2c3

  Copy full SHA for 5bcd2c3

• TorchFX: IsFinite op support

  cavusmustafa

  committed
  ea90465

  Copy full SHA for ea90465

• aten.hardtanh unit test support for multiple data types and input shapes

  cavusmustafa

  committed
  3bc2b47

  Copy full SHA for 3bc2b47

Commits on Mar 11, 2024

• TorchFX: aten.any translation uptade

  cavusmustafa

  committed
  4b7ce29

  Copy full SHA for 4b7ce29

• Merge branch 'master' into torch_compile/new_op_support_3

  cavusmustafa

  authored
  2f025a5

  Copy full SHA for 2f025a5

• Simplified aten.any translation

  cavusmustafa

  committed
  c1544e1

  Copy full SHA for c1544e1

Commits on Mar 12, 2024

• TorchFX: Support for aten.any.dim

  cavusmustafa

  committed
  6e49c99

  Copy full SHA for 6e49c99

• TorchFX: Removed converting to i64 in sort translation

  cavusmustafa

  committed
  3f33267

  Copy full SHA for 3f33267

• Sort translation formatting fix

  cavusmustafa

  committed
  6036867

  Copy full SHA for 6036867

Commits on Mar 13, 2024

• New unit tests only enabled for FX backend

  cavusmustafa

  committed
  6e344e5

  Copy full SHA for 6e344e5

• TorchFX: split unit test temporarily disabled

  cavusmustafa

  committed
  8652d61

  Copy full SHA for 8652d61

Commits on Mar 14, 2024

• scale support for elu translation & typo fix

  cavusmustafa

  committed
  030aba4

  Copy full SHA for 030aba4

• Removed bitwise_right_shift for FX

  cavusmustafa

  committed
  5aaf117

  Copy full SHA for 5aaf117

• Merge branch 'master' into torch_compile/new_op_support_3

  cavusmustafa

  authored
  c59c716

  Copy full SHA for c59c716

• Removed unused include lines in elu

  cavusmustafa

  committed
  51deae6

  Copy full SHA for 51deae6

• Code formatting fix

  cavusmustafa

  committed
  cb68bbe

  Copy full SHA for cb68bbe

Commits on Mar 18, 2024

• TorchFX: Log sigmoid fix

  cavusmustafa

  committed
  2e1becf

  Copy full SHA for 2e1becf

Commits on Mar 19, 2024

• Update src/frontends/pytorch/src/op/any.cpp

  Show description for 5177557
  

  cavusmustafa

  and

  mvafin

  authored
  5177557

  Copy full SHA for 5177557

• Update tests/layer_tests/pytorch_tests/test_unary_ops.py

  Show description for ae27838
  

  cavusmustafa

  and

  mvafin

  authored
  ae27838

  Copy full SHA for ae27838

• Merge branch 'master' into torch_compile/new_op_support_3

  cavusmustafa

  authored
  e4d1f0e

  Copy full SHA for e4d1f0e

• Any translation fix

  cavusmustafa

  committed
  de5e913

  Copy full SHA for de5e913

• Enable TorchFX unit test for new_ones

  cavusmustafa

  committed
  4f28e19

  Copy full SHA for 4f28e19

• argmax_argmin translation using stable=true

  cavusmustafa

  committed
  d7c50db

  Copy full SHA for d7c50db

• Merge branch 'master' into torch_compile/new_op_support_3

  cavusmustafa

  authored
  d8fcf9e

  Copy full SHA for d8fcf9e

• Code formatting fix

  cavusmustafa

  committed
  bffe432

  Copy full SHA for bffe432

• Code formatting fix

  cavusmustafa

  committed
  4c25f5a

  Copy full SHA for 4c25f5a

• Code formatting fix

  cavusmustafa

  committed
  25a5380

  Copy full SHA for 25a5380

• Code formatting fix

  cavusmustafa

  committed
  611b61e

  Copy full SHA for 611b61e

• Revert removing convert from argmax_argmin translation

  cavusmustafa

  committed
  aab368c

  Copy full SHA for aab368c

• Code formatting fix

  cavusmustafa

  committed
  e06489c

  Copy full SHA for e06489c

• Code formatting fix

  cavusmustafa

  committed
  01e1d85

  Copy full SHA for 01e1d85

Commits on Mar 20, 2024

• argmax_argmin topk fix

  cavusmustafa

  committed
  0b47a86

  Copy full SHA for 0b47a86

• Merge branch 'master' into torch_compile/new_op_support_3

  cavusmustafa

  authored
  adaad2d

  Copy full SHA for adaad2d

• TorchFX: revert changes to use new new_ones, new_full, and ones translation

  cavusmustafa

  committed
  86bf932

  Copy full SHA for 86bf932

• Merge branch 'master' into torch_compile/new_op_support_3

  cavusmustafa

  authored
  f10afed

  Copy full SHA for f10afed

• Added div_ and rand ops

  suryasidd

  committed
  20d2afa

  Copy full SHA for 20d2afa

• log_sigmoid fix for FX and TS

  cavusmustafa

  committed
  293493c

  Copy full SHA for 293493c

• Added missing div_ to op_table

  suryasidd

  committed
  745a451

  Copy full SHA for 745a451

• Merge branch 'master' into torch_compile/new_op_support_3

  cavusmustafa

  authored
  92af4e2

  Copy full SHA for 92af4e2