diff --git a/components/heap/CMakeLists.txt b/components/heap/CMakeLists.txt
index a05c1bc4db..ab5f978eee 100644
--- a/components/heap/CMakeLists.txt
+++ b/components/heap/CMakeLists.txt
@@ -42,7 +42,7 @@ idf_component_register(SRCS "${srcs}"
                     INCLUDE_DIRS ${includes}
                     PRIV_INCLUDE_DIRS ${priv_includes}
                     LDFRAGMENTS linker.lf
-                    PRIV_REQUIRES soc)
+                    PRIV_REQUIRES soc spi_flash)
 
 if(CONFIG_HEAP_TRACING)
     set(WRAP_FUNCTIONS
diff --git a/components/heap/heap_caps.c b/components/heap/heap_caps.c
index 9c35b56a81..50fe13763d 100644
--- a/components/heap/heap_caps.c
+++ b/components/heap/heap_caps.c
@@ -14,7 +14,7 @@
 #include "esp_log.h"
 #include "heap_private.h"
 #include "esp_system.h"
-
+#include "esp_private/cache_utils.h"
 
 /* Forward declaration for base function, put in IRAM.
  * These functions don't check for errors after trying to allocate memory. */
@@ -56,10 +56,16 @@ IRAM_ATTR static void *dram_alloc_to_iram_addr(void *addr, size_t len)
 }
 
 
-static void heap_caps_alloc_failed(size_t requested_size, uint32_t caps, const char *function_name)
+IRAM_ATTR static void heap_caps_alloc_failed(size_t requested_size, uint32_t caps, const char *function_name)
 {
+    static const DRAM_ATTR char *default_func_name = "<function_name>";
     if (alloc_failed_callback) {
-        alloc_failed_callback(requested_size, caps, function_name);
+        if (!spi_flash_cache_enabled() && !esp_ptr_internal(function_name)) {
+            alloc_failed_callback(requested_size, caps, default_func_name);
+        }
+        else {
+            alloc_failed_callback(requested_size, caps, function_name);
+        }
     }
 
 #ifdef CONFIG_HEAP_ABORT_WHEN_ALLOCATION_FAILS
diff --git a/components/heap/heap_private.h b/components/heap/heap_private.h
index 51b6773ffa..e7b8683229 100644
--- a/components/heap/heap_private.h
+++ b/components/heap/heap_private.h
@@ -43,7 +43,7 @@ extern SLIST_HEAD(registered_heap_ll, heap_t_) registered_heaps;
 bool heap_caps_match(const heap_t *heap, uint32_t caps);
 
 /* return all possible capabilities (across all priorities) for a given heap */
-inline static uint32_t get_all_caps(const heap_t *heap)
+inline static __attribute__((always_inline)) uint32_t get_all_caps(const heap_t *heap)
 {
     if (heap->heap == NULL) {
         return 0;
diff --git a/components/heap/test/CMakeLists.txt b/components/heap/test/CMakeLists.txt
index 0817b74f49..2d17b5d996 100644
--- a/components/heap/test/CMakeLists.txt
+++ b/components/heap/test/CMakeLists.txt
@@ -2,3 +2,17 @@ idf_component_register(SRC_DIRS "."
                     PRIV_INCLUDE_DIRS "."
                     PRIV_REQUIRES cmock test_utils heap spi_flash)
 target_compile_options(${COMPONENT_LIB} PRIVATE "-Wno-format")
+
+if(CONFIG_COMPILER_DUMP_RTL_FILES)
+    idf_build_get_property(elf_file_name EXECUTABLE GENERATOR_EXPRESSION)
+    add_custom_target(check_test_app_sections ALL
+                      COMMAND ${PYTHON} $ENV{IDF_PATH}/tools/ci/check_callgraph.py
+                      --rtl-dir ${CMAKE_BINARY_DIR}/esp-idf/heap/
+                      --elf-file ${CMAKE_BINARY_DIR}/${elf_file_name}
+                      find-refs
+                      --from-sections=.iram0.text
+                      --to-sections=.flash.text,.flash.rodata
+                      --exit-code
+                      DEPENDS ${elf_file_name}
+                      )
+endif()
diff --git a/tools/unit-test-app/sdkconfig.defaults b/tools/unit-test-app/sdkconfig.defaults
index 486727de5a..af3f9a19bd 100644
--- a/tools/unit-test-app/sdkconfig.defaults
+++ b/tools/unit-test-app/sdkconfig.defaults
@@ -31,3 +31,4 @@ CONFIG_MQTT_TEST_BROKER_URI="mqtt://${EXAMPLE_MQTT_BROKER_TCP}"
 # See esp_spiffs_gc description for more info.
 CONFIG_SPIFFS_GC_MAX_RUNS=132
 CONFIG_NVS_ASSERT_ERROR_CHECK=y
+CONFIG_COMPILER_DUMP_RTL_FILES=y