代码拉取完成,页面将自动刷新
2239
Add auto parallel cost model and unittests
已合并
PR types
New features
PR changes
Others
Describe
add cost model for distributed program generated by auto parallel module
- estimate_cost(distributed_program, cluster, process_mesh, single_cost_data, batch_size): this function takes the distributed program, cluster model and other distributed settings as inputs and estimate the runtime of one iteration and static memory and peak memory consumption.