mlkem test diagnostics and allocator updates

sdk/software/examples/mlkem_test/sw_mlkem_config.h

@@ -11,11 +11,19 @@
 #include "mlkem_native_config.h"
 
 #define MLK_CONFIG_CUSTOM_ZEROIZE
+#define MLK_CONFIG_CUSTOM_ALLOC_FREE
 
 #if !defined(__ASSEMBLER__)
 #include <stddef.h>
 #include <stdint.h>
 #include "src/sys.h"
+#include "sw_mlkem_alloc.h"
+
+#define MLK_CUSTOM_ALLOC(v, T, N) \
+    T *(v) = (T *)sw_mlkem_alloc(sizeof(T) * (N))
+
+#define MLK_CUSTOM_FREE(v, T, N) \
+    sw_mlkem_free((void *)(v), sizeof(T) * (N))
 
 static MLK_INLINE void mlk_zeroize(void *ptr, size_t len)
 {