fix segmentation fault on temporary buffer

v923z · Jan 28, 2024 · 30c7f4f · 30c7f4f
1 parent feb9ad9
commit 30c7f4f
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/code/utils/utils.c b/code/utils/utils.c
@@ -367,10 +367,13 @@ mp_obj_t utils_spectrogram(size_t n_args, const mp_obj_t *pos_args, mp_map_t *kw
         spectrum += spectrum_sz;
     }
 
-    // FIXME:
-    // if(scratchpad_object == mp_const_none) {
-    //     m_del(mp_float_t, tmp, 2 * len);
-    // }
+    if(scratchpad_object == mp_const_none) {
+        tmp -= len;
+        #if ULAB_FFT_IS_NUMPY_COMPATIBLE
+        tmp -= len;
+        #endif
+        m_del(mp_float_t, tmp, 2 * len);
+    }
     return MP_OBJ_FROM_PTR(out);
 }