MCPcopy Create free account
hub / github.com/MegEngine/MegCC / ArgSortDeduceLayout

Class ArgSortDeduceLayout

compiler/lib/KernelGen/Common/DeduceLayoutMap.cpp:126–142  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

124 }
125};
126class ArgSortDeduceLayout : public DeduceFunc {
127public:
128 std::string GetDeduceSymbol(TContext* context) const override {
129 return "DeduceFunc_ArgSort";
130 }
131 std::string GetDeduceBody(TContext* context) const override {
132 std::stringstream writer;
133 writer << GenCommonRet() << " " << GetDeduceSig(context) << "{\n";
134 writer << R"(
135 outputs[0]->layout = inputs[0]->layout;
136 outputs[1]->layout = inputs[0]->layout;
137 return TinyNN_SUCCESS;
138 }
139 )";
140 return writer.str();
141 }
142};
143
144class ConcatDeduceLayout : public DeduceFunc {
145public:

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected