diff --git a/python/paddle/fluid/tests/custom_op/custom_raw_op_kernel_op_setup.py b/python/paddle/fluid/tests/custom_op/custom_raw_op_kernel_op_setup.py
index e751a335d72319de71c95433dabb39e2e65562b4..a1ae927997200c9bb796d60bf84510988409c213 100644
--- a/python/paddle/fluid/tests/custom_op/custom_raw_op_kernel_op_setup.py
+++ b/python/paddle/fluid/tests/custom_op/custom_raw_op_kernel_op_setup.py
@@ -38,6 +38,7 @@ if core.is_compiled_with_mkldnn():
     macros.append(("PADDLE_WITH_MKLDNN", None))
 if core.is_compiled_with_nccl():
     macros.append(("PADDLE_WITH_NCCL", None))
+macros.append(("THRUST_IGNORE_CUB_VERSION_CHECK", None))
 
 include_dirs = list(paddle_includes) + [cwd]
 setup(name=os.getenv("MODULE_NAME", "custom_raw_op_kernel_op_setup"),