xla
Add fx passes to support unbounded dynamism in torch op arg
#6653

Merged

Add fx passes to support unbounded dynamism in torch op arg #6653

qihqi merged 9 commits into master from lsiyuan/export-dynamism-fx-pass

lsy323 force pushed from 531d34b3 to f6a6d469 1 year ago

lsy323 force pushed from f6a6d469 to 8faf3939 1 year ago

lsy323 force pushed from 8507793c to 85a6029c 1 year ago

lsy323 force pushed from b7924af4 to de636acb 1 year ago

lsy323 changed the title ~~[WIP] Add some fx passes to support exporting unbounded dynamism~~ Add some fx passes to support exporting unbounded dynamism 1 year ago

lsy323 force pushed from de636acb to 6c237e65 1 year ago

lsy323 force pushed from 6c237e65 to 4b416598 1 year ago

lsy323 changed the title ~~Add some fx passes to support exporting unbounded dynamism~~ Add fx passes to support exporting unbounded dynamism 1 year ago

add unbounded dynamism test for some aten ops

5965d021

remove export script

a709c4d4

undo support to view and softmax

ff2671b9

remove debug print

76016476

disable unsupported tests

3e3a1657

add comments to shape inference

56613b74

fix linter

3cfd8f2c

fix test util path

ce436f89

lsy323 force pushed from 4b416598 to ce436f89 1 year ago

yapf

391dbf70

lsy323 requested a review from

qihqi 1 year ago

lsy323 added dynamism

qihqi approved these changes on 2024-03-20

qihqi merged 7e0d3a5a into master 1 year ago

lsy323 deleted the lsiyuan/export-dynamism-fx-pass branch 1 year ago

lsy323 changed the title ~~Add fx passes to support exporting unbounded dynamism~~ Add fx passes to support unbounded dynamism in torch op arg 1 year ago

miladm assigned

lsy323 1 year ago

Reviewers

qihqi

Assignees

lsy323

Labels

dynamism

Milestone

No milestone

xla Add fx passes to support unbounded dynamism in torch op arg #6653 Merged

Add fx passes to support unbounded dynamism in torch op arg #6653

xla
Add fx passes to support unbounded dynamism in torch op arg
#6653

Merged