AOT/JIT native stack bound check improvement (#2244)
Move the native stack overflow check from the caller to the callee because the former doesn't work for call_indirect and imported functions. Make the stack usage estimation more accurate. Instead of making a guess from the number of wasm locals in the function, use the LLVM's idea of the stack size of each MachineFunction. The former is inaccurate because a) it doesn't reflect optimization passes, and b) wasm locals are not the only reason to use stack. To use the post-compilation stack usage information without requiring 2-pass compilation or machine-code imm rewriting, introduce a global array to store stack consumption of each functions: For JIT, use a custom IRCompiler with an extra pass to fill the array. For AOT, use `clang -fstack-usage` equivalent because we support external llc. Re-implement function call stack usage estimation to reflect the real calling conventions better. (aot_estimate_stack_usage_for_function_call) Re-implement stack estimation logic (--enable-memory-profiling) based on the new machinery. Discussions: #2105.
This commit is contained in:
@ -19,6 +19,15 @@ extern "C" {
|
||||
#define AOT_FUNC_PREFIX "aot_func#"
|
||||
#endif
|
||||
|
||||
#ifndef AOT_FUNC_INTERNAL_PREFIX
|
||||
#define AOT_FUNC_INTERNAL_PREFIX "aot_func_internal#"
|
||||
#endif
|
||||
|
||||
#ifndef AOT_STACK_SIZES_NAME
|
||||
#define AOT_STACK_SIZES_NAME "aot_stack_sizes"
|
||||
#endif
|
||||
extern const char *aot_stack_sizes_name;
|
||||
|
||||
typedef InitializerExpression AOTInitExpr;
|
||||
typedef WASMType AOTFuncType;
|
||||
typedef WASMExport AOTExport;
|
||||
|
||||
Reference in New Issue
Block a user