MCPcopy Create free account
hub / github.com/Oneflow-Inc/oneflow / comp

Class comp

oneflow/core/auto_parallel/auto_memory.cpp:83–94  ·  view source on GitHub ↗

Order in the waiting sets

Source from the content-addressed store, hash-verified

81
82// Order in the waiting sets
83struct comp {
84 bool operator()(const TopoStruct* a, const TopoStruct* b) const {
85 for (auto decide_parameter : decide_parameters) {
86 auto decide_parameter_a = a->GetDecidingParameter(decide_parameter);
87 auto decide_parameter_b = b->GetDecidingParameter(decide_parameter);
88 if (decide_parameter_a != decide_parameter_b) {
89 return decide_parameter_a < decide_parameter_b;
90 }
91 }
92 return a->op_node->op().op_name() < b->op_node->op().op_name();
93 }
94};
95
96bool IsProducedRegisterReusable(const Operator& op) {
97 // The repeat, acc, pack and unpack operators have non-reusable registers

Callers 1

DecideOrderFunction · 0.70

Calls

no outgoing calls

Tested by

no test coverage detected