diff --git a/paddle/phi/backends/device_ext.h b/paddle/phi/backends/device_ext.h
index 6315fe15afdf1ecd9c7657396468320eda7d88c1..749d8d323b62d1cfd7429f0f2b20da94de8e956f 100644
--- a/paddle/phi/backends/device_ext.h
+++ b/paddle/phi/backends/device_ext.h
@@ -523,14 +523,14 @@ struct CustomRuntimeParams {
   char reserved[32];
 };
 
-#define PADDLE_CUSTOM_RUNTIME_CHECK_VERSION(params)             \
-  if ((params)->size != sizeof(DevicePluginParams) &&           \
-      (params)->interface->size != sizeof(C_DeviceInterface)) { \
-    return;                                                     \
-  }                                                             \
-  (params)->version.major = PADDLE_DEVICE_PLUGIN_MAJOR_VERSION; \
-  (params)->version.minor = PADDLE_DEVICE_PLUGIN_MINOR_VERSION; \
-  (params)->version.patch = PADDLE_DEVICE_PLUGIN_PATCH_VERSION;
+#define PADDLE_CUSTOM_RUNTIME_CHECK_VERSION(params)              \
+  if ((params)->size != sizeof(CustomRuntimeParams) &&           \
+      (params)->interface->size != sizeof(C_DeviceInterface)) {  \
+    return;                                                      \
+  }                                                              \
+  (params)->version.major = PADDLE_CUSTOM_RUNTIME_MAJOR_VERSION; \
+  (params)->version.minor = PADDLE_CUSTOM_RUNTIME_MINOR_VERSION; \
+  (params)->version.patch = PADDLE_CUSTOM_RUNTIME_PATCH_VERSION;
 
 // Plugin implement it and fill CustomRuntimeParams
 void InitPlugin(CustomRuntimeParams*);