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

Function GenCommonWorkspaceCall

compiler/lib/KernelGen/KernelGen.cpp:139–142  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

137 int* nr_out_weight, const RuntimeOpt* opt))";
138}
139std::string GenCommonWorkspaceCall() {
140 return R"((
141 Tensor** inputs, int nr_input, int nr_thread, size_t* workspace))";
142}
143std::string GenCommonDeduceCall() {
144 return R"((
145 Tensor** inputs, int nr_input, Tensor** outputs, int nr_output))";

Callers 2

get_helper_module_strMethod · 0.85
GetWorkspaceSignatureMethod · 0.85

Calls

no outgoing calls

Tested by 1

get_helper_module_strMethod · 0.68