| 224 | } |
| 225 | extern std::unordered_map<std::string, StdKernelWorkspaceCall> g_str_2_kern_workspace; |
| 226 | StdKernelWorkspaceCall TargetModule::get_kernel_workspace(std::string& sig) const { |
| 227 | #if !MEGCC_TEST_GEN |
| 228 | auto res = g_str_2_kern_workspace.find(sig); |
| 229 | if (res == g_str_2_kern_workspace.end()) { |
| 230 | return nullptr; |
| 231 | } else { |
| 232 | return res->second; |
| 233 | } |
| 234 | #else |
| 235 | return nullptr; |
| 236 | #endif |
| 237 | } |
| 238 | |
| 239 | extern std::unordered_map<std::string, size_t> g_str_2_workspace_size; |
| 240 | size_t TargetModule::get_kernel_workspace_size(std::string& sig) const { |