diff --git a/osal/linux/os_allocator.c b/osal/linux/os_allocator.c
index 72fdc643..532ed21c 100644
--- a/osal/linux/os_allocator.c
+++ b/osal/linux/os_allocator.c
@@ -35,7 +35,11 @@ MPP_RET os_allocator_get(os_allocator *api, MppBufferType type)
     } break;
     case MPP_BUFFER_TYPE_ION : {
 #ifdef RKPLATFORM
+#ifdef HAVE_DRM
+        *api = allocator_drm;
+#else
         *api = allocator_ion;
+#endif
 #else
         *api = allocator_std;
 #endif