Skip adding reshapes for resharding tensors where their dimension sizes are...
Skip adding reshapes for resharding tensors where their dimension sizes are not divisible by the mesh dimensions. Also get rid of a duplicate debug print statement PiperOrigin-RevId: 623571611
显示
- third_party/xla/xla/hlo/experimental/auto_sharding/auto_sharding.cc 21 个添加, 20 个删除...y/xla/xla/hlo/experimental/auto_sharding/auto_sharding.cc
- third_party/xla/xla/hlo/experimental/auto_sharding/auto_sharding_util.cc 31 个添加, 26 个删除.../xla/hlo/experimental/auto_sharding/auto_sharding_util.cc
- third_party/xla/xla/hlo/experimental/auto_sharding/auto_sharding_util.h 3 个添加, 5 个删除...a/xla/hlo/experimental/auto_sharding/auto_sharding_util.h
想要评论请 注册 或 登录