diff --git a/zephyr/lib/alloc.c b/zephyr/lib/alloc.c
index 1e6d6a733..4a6ea68cf 100644
--- a/zephyr/lib/alloc.c
+++ b/zephyr/lib/alloc.c
@@ -52,7 +52,7 @@ extern struct tr_ctx zephyr_tr;
 
 /* The Zephyr heap */
 
-#ifdef CONFIG_IMX
+#if defined(CONFIG_IMX) || defined(CONFIG_AMD)
 
 #ifdef CONFIG_XTENSA
 #define HEAPMEM_SIZE		(HEAP_SYSTEM_SIZE + HEAP_RUNTIME_SIZE + HEAP_BUFFER_SIZE)