PR #46103 [Fix] Remove calls to unsupported functions in TP

[Fix] Remove calls to unsupported functions in TP #46103

ArthurZucker merged 4 commits into main from fix-tp-unsupported

Alias without _ to be compatible with torch < 2.10

8a74e7fa

Remove the _unflatten call which is not compatible w/ torch < 2.9.1

28d90482

Add a TODO and remove an assert

330ac560

remi-or requested a review from

3outeille 56 days ago

remi-or requested a review from

ArthurZucker 56 days ago

remi-or assigned

remi-or 56 days ago

3outeille approved these changes on 2026-05-20

Merge branch 'main' into fix-tp-unsupported

dc7efcbd

ArthurZucker approved these changes on 2026-05-20

ArthurZucker merged 4dbb3587 into main 56 days ago

ArthurZucker deleted the fix-tp-unsupported branch 56 days ago

Reviewers

ArthurZucker

3outeille

Assignees

remi-or

Labels

None yet

Milestone

No milestone