Stabilize wasm runtime-cache tests by clearing in-process JIT cache

Author: FrancescAlted

src/dsl_jit_backend_wasm32.c

@@ -511,6 +511,19 @@ typedef struct {
 static me_dsl_jit_wasm_pos_cache_entry g_dsl_jit_wasm_pos_cache[ME_DSL_JIT_WASM_POS_CACHE_SLOTS];
 static int g_dsl_jit_wasm_pos_cache_cursor = 0;
 
+void dsl_jit_wasm_pos_cache_reset_for_tests(void) {
+    for (int i = 0; i < ME_DSL_JIT_WASM_POS_CACHE_SLOTS; i++) {
+        if (g_dsl_jit_wasm_pos_cache[i].valid) {
+            if (g_dsl_jit_wasm_pos_cache[i].fn_idx != 0) {
+                dsl_wasm_jit_free_dispatch(g_dsl_jit_wasm_pos_cache[i].fn_idx);
+            }
+            free(g_dsl_jit_wasm_pos_cache[i].scratch);
+            memset(&g_dsl_jit_wasm_pos_cache[i], 0, sizeof(g_dsl_jit_wasm_pos_cache[i]));
+        }
+    }
+    g_dsl_jit_wasm_pos_cache_cursor = 0;
+}
+
 static int dsl_jit_wasm_pos_cache_find_slot(uint64_t key) {
     for (int i = 0; i < ME_DSL_JIT_WASM_POS_CACHE_SLOTS; i++) {
         if (g_dsl_jit_wasm_pos_cache[i].valid && g_dsl_jit_wasm_pos_cache[i].key == key) {

src/dsl_jit_runtime_internal.h

@@ -263,6 +263,7 @@ bool dsl_jit_cc_promote_self_symbols_global(void);
 #if ME_USE_WASM32_JIT
 bool dsl_jit_compile_wasm32(me_dsl_compiled_program *program, uint64_t key);
 bool dsl_jit_wasm_pos_cache_bind_program(me_dsl_compiled_program *program, uint64_t key);
+void dsl_jit_wasm_pos_cache_reset_for_tests(void);
 void dsl_wasm_jit_free_dispatch(int idx);
 void dsl_register_wasm_jit_helpers(me_wasm_jit_instantiate_helper instantiate_helper,
                                    me_wasm_jit_free_helper free_helper);

tests/test_dsl_jit_runtime_cache.c

@@ -2151,6 +2151,7 @@ static int test_wasm_reserved_index_cache_key_differentiation(void) {
         printf("  FAILED: setenv ME_DSL_JIT=1 failed\n");
         goto cleanup;
     }
+    dsl_jit_wasm_pos_cache_reset_for_tests();
     test_wasm_runtime_cache_reset_instantiate_count();
 
     if (me_compile(src_global, NULL, 0, ME_FLOAT64, &err, &expr) != ME_COMPILE_SUCCESS || !expr) {
@@ -2264,6 +2265,7 @@ static int test_wasm_runtime_cache_eviction_policy(void) {
         printf("  FAILED: setenv ME_DSL_JIT=1 failed\n");
         goto cleanup;
     }
+    dsl_jit_wasm_pos_cache_reset_for_tests();
     test_wasm_runtime_cache_reset_instantiate_count();
 
     if (snprintf(src, sizeof(src),
@@ -2482,14 +2484,17 @@ static int test_wasm_instantiate_warning_policy(void) {
     printf("\n=== DSL JIT Runtime Cache Test: wasm instantiate warning policy ===\n");
 
     static const char *src_default =
+        "# me:compiler=cc\n"
         "def kernel_warn_default(x):\n"
         "    y = x + 41.0\n"
         "    return y\n";
     static const char *src_on =
+        "# me:compiler=cc\n"
         "def kernel_warn_on(x):\n"
         "    y = x + 42.0\n"
         "    return y\n";
     static const char *src_trace =
+        "# me:compiler=cc\n"
         "def kernel_warn_trace(x):\n"
         "    y = x + 43.0\n"
         "    return y\n";