From 462cf0ef28057981940cc2feccfc5892f72c9a2c Mon Sep 17 00:00:00 2001
From: sneaxiy <32832641+sneaxiy@users.noreply.github.com>
Date: Mon, 5 Sep 2022 19:20:10 +0800
Subject: [PATCH] fix raw kernel op (#45733)

---
 .../fluid/tests/custom_op/custom_raw_op_kernel_op_setup.py       | 1 +
 1 file changed, 1 insertion(+)

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 e751a335d7..a1ae927997 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"),
-- 
GitLab