search

mlc-ai / relax

Apache License 2.0
149 stars 75 forks source link

[DistIR] Support constant DTensor #259

Closed jinhongyii closed 1 year ago

jinhongyii commented 1 year ago
  1. enable constant to be DTensor (replica only, no sharding)
  2. support constant in sharding propagation
  3. add test for dynamic shape model cc: @tqchen