christoph/wamr
1
Fork 0
Code Releases Activity

Enable static PGO for Linux SGX (#2270)

Browse Source 
Enable static PGO for Linux SGX and update the related benchmarks
test scripts and documents.
This commit is contained in:
TianlongLiang
2023-06-09 14:13:43 +08:00
committed by GitHub
parent cabcb177c8
commit 2f01cb7b7a
15 changed files with 209 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

82
product-mini/platforms/linux-sgx/enclave-sample/App/App.cpp
View File

@ -232,6 +232,9 @@ print_help()
printf(" for example:\n");
printf(" --addr-pool=1.2.3.4/15,2.3.4.5/16\n");
printf(" --max-threads=n Set maximum thread number per cluster, default is 4\n");
#if WASM_ENABLE_STATIC_PGO != 0
printf(" --gen-prof-file=<path> Generate LLVM PGO (Profile-Guided Optimization) profile file\n");
#endif
printf(" --version Show version information\n");
return 1;
}
@ -294,6 +297,10 @@ typedef enum EcallCmd {
CMD_SET_WASI_ARGS, /* wasm_runtime_set_wasi_args() */
CMD_SET_LOG_LEVEL, /* bh_log_set_verbose_level() */
CMD_GET_VERSION, /* wasm_runtime_get_version() */
#if WASM_ENABLE_STATIC_PGO != 0
CMD_GET_PGO_PROF_BUF_SIZE, /* wasm_runtime_get_pro_prof_data_size() */
CMD_DUMP_PGO_PROF_BUF_DATA, /* wasm_runtime_dump_pgo_prof_data_to_buf() */
#endif
} EcallCmd;
static void
@ -598,6 +605,64 @@ get_version(uint64_t *major, uint64_t *minor, uint64_t *patch)
*patch = ecall_args[2];
}
#if WASM_ENABLE_STATIC_PGO != 0
static void
dump_pgo_prof_data(void *module_inst, const char *path)
{
char *buf;
uint32_t len;
FILE *file;
uint64_t ecall_args[1];
ecall_args[0] = (uint64_t)(uintptr_t)module_inst;
if (SGX_SUCCESS
!= ecall_handle_command(g_eid, CMD_GET_PGO_PROF_BUF_SIZE,
(uint8_t *)ecall_args, sizeof(ecall_args))) {
printf("Call ecall_handle_command() failed.\n");
return;
}
if (!(len = ecall_args[0])) {
printf("failed to get LLVM PGO profile data size\n");
return;
}
if (!(buf = (char *)malloc(len))) {
printf("allocate memory failed\n");
return;
}
uint64_t ecall_args_2[3];
ecall_args_2[0] = (uint64_t)(uintptr_t)module_inst;
ecall_args_2[1] = (uint64_t)(uintptr_t)buf;
ecall_args_2[2] = len;
if (SGX_SUCCESS
!= ecall_handle_command(g_eid, CMD_DUMP_PGO_PROF_BUF_DATA,
(uint8_t *)ecall_args_2,
sizeof(ecall_args_2))) {
printf("Call ecall_handle_command() failed.\n");
free(buf);
return;
}
if (!(len = ecall_args_2[0])) {
printf("failed to dump LLVM PGO profile data\n");
free(buf);
return;
}
if (!(file = fopen(path, "wb"))) {
printf("failed to create file %s", path);
free(buf);
return;
}
fwrite(buf, len, 1, file);
fclose(file);
free(buf);
printf("LLVM raw profile file %s was generated.\n", path);
}
#endif
int
main(int argc, char *argv[])
{
@ -619,6 +684,9 @@ main(int argc, char *argv[])
const char *addr_pool[8] = { NULL };
uint32_t addr_pool_size = 0;
uint32_t max_thread_num = 4;
#if WASM_ENABLE_STATIC_PGO != 0
const char *gen_prof_file = NULL;
#endif
if (enclave_init(&g_eid) < 0) {
std::cout << "Fail to initialize enclave." << std::endl;
@ -718,6 +786,13 @@ main(int argc, char *argv[])
return print_help();
max_thread_num = atoi(argv[0] + 14);
}
#if WASM_ENABLE_STATIC_PGO != 0
else if (!strncmp(argv[0], "--gen-prof-file=", 16)) {
if (argv[0][16] == '\0')
return print_help();
gen_prof_file = argv[0] + 16;
}
#endif
else if (!strncmp(argv[0], "--version", 9)) {
uint64_t major = 0, minor = 0, patch = 0;
get_version(&major, &minor, &patch);
@ -779,6 +854,11 @@ main(int argc, char *argv[])
else
app_instance_main(wasm_module_inst, argc, argv);
#if WASM_ENABLE_STATIC_PGO != 0
if (gen_prof_file)
dump_pgo_prof_data(wasm_module_inst, gen_prof_file);
#endif
ret = 0;
/* Deinstantiate module */
@ -836,7 +916,7 @@ wamr_pal_create_process(struct wamr_pal_create_process_args *args)
int stdoutfd = -1;
int stderrfd = -1;
int argc = 2;
const int argc = 2;
char *argv[argc] = { (char *)"./iwasm", (char *)args->argv[0] };
uint8_t *wasm_files_buf = NULL;