From 2110e6e25af9ca52d4dc6a9357e9c7f3c5f1c5ef Mon Sep 17 00:00:00 2001
From: ckl117 <ckl117@163.com>
Date: Tue, 18 Nov 2025 18:35:49 +0800
Subject: [PATCH] check METAX_GPU

---
 custom_ops/gpu_ops/helper.h | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/custom_ops/gpu_ops/helper.h b/custom_ops/gpu_ops/helper.h
index b06ec0211eb..5e41d122796 100644
--- a/custom_ops/gpu_ops/helper.h
+++ b/custom_ops/gpu_ops/helper.h
@@ -724,6 +724,10 @@ inline void launchWithPdlWhenEnabled(KernelFn kernelFn,
                                      size_t dynamicShmSize,
                                      cudaStream_t stream,
                                      Args &&...args) {
+#ifdef PADDLE_WITH_CUSTOM_DEVICE_METAX_GPU
+  (*kernelFn)<<<grid, block, dynamicShmSize, stream>>>(
+      std::forward<Args>(args)...);
+#else
   cudaLaunchConfig_t kernelConfig;
   kernelConfig.gridDim = grid;
   kernelConfig.blockDim = block;
@@ -737,4 +741,5 @@ inline void launchWithPdlWhenEnabled(KernelFn kernelFn,
   kernelConfig.numAttrs = 1;
 
   cudaLaunchKernelEx(&kernelConfig, kernelFn, std::forward<Args>(args)...);
+#endif
 }