diff --git a/include/my_compiler.h b/include/my_compiler.h
index ff049f3d8a6..831929cf5ab 100644
--- a/include/my_compiler.h
+++ b/include/my_compiler.h
@@ -156,6 +156,7 @@ program.  The paths leading to call of cold functions within code are
 marked as unlikely by the branch prediction mechanism.  optimize a
 rarely invoked function for size instead for speed. */
 # define ATTRIBUTE_COLD __attribute__((cold))
+# define ATTRIBUTE_MALLOC __attribute__((malloc))
 #elif defined _MSC_VER
 # define ATTRIBUTE_NORETURN __declspec(noreturn)
 # define ATTRIBUTE_NOINLINE __declspec(noinline)
@@ -168,6 +169,10 @@ rarely invoked function for size instead for speed. */
 # define ATTRIBUTE_COLD /* empty */
 #endif
 
+#ifndef ATTRIBUTE_MALLOC
+# define ATTRIBUTE_MALLOC
+#endif
+
 #include <my_attribute.h>
 
 #endif /* MY_COMPILER_INCLUDED */
diff --git a/mysys/my_malloc.c b/mysys/my_malloc.c
index a1a7a1303df..a2371ea1ad0 100644
--- a/mysys/my_malloc.c
+++ b/mysys/my_malloc.c
@@ -62,6 +62,7 @@ void set_malloc_size_cb(MALLOC_SIZE_CB func)
 
   @return A pointer to the allocated memory block, or NULL on failure.
 */
+ATTRIBUTE_MALLOC
 void *my_malloc(PSI_memory_key key, size_t size, myf my_flags)
 {
   my_memory_header *mh;