Enable dynamic shapes for pipeline parallel engine inputs #5481
loadams
merged
18 commits into
deepspeedai:master
from
tohtana:tohtana/pp_dynamic_shape
allow dynamic shape of inputs to pp engine
679ac4e2
Merge branch 'master' into tohtana/pp_dynamic_shape
3e15fae0
improve buffer allocation
b3c07504
fix condition to extend cache
822aeeef
consolidate communication for tensor metadata
9f96ad40
Merge branch 'master' into tohtana/pp_dynamic_shape
c47ef5a7
Merge branch 'microsoft:master' into tohtana/pp_dynamic_shape
85633e63
Merge branch 'master' into tohtana/pp_dynamic_shape
652c148f
Merge branch 'master' into tohtana/pp_dynamic_shape
428dbe19
fix error after auto merge
4c7910bd
tjruwase
commented
on 2024-07-25
tjruwase
commented
on 2024-07-25
Merge branch 'master' into tohtana/pp_dynamic_shape
efa36b39
add test for dynamic shape support
b6aecacc
refactor duplicated code
cb99e11a
remove unreachable code
bab2247b
add assertions
e9b03cfd
add comments and remove unused method
7fe47d0e
Merge branch 'master' into tohtana/pp_dynamic_shape
33453bf2
Merge branch 'master' into tohtana/pp_dynamic_shape
99257826
tjruwase
approved these changes
on 2024-08-16
Login to write a write a comment.
Login via GitHub