未验证 提交 462cf0ef 编写于 作者: S sneaxiy 提交者: GitHub

fix raw kernel op (#45733)

上级 a8da1625
...@@ -38,6 +38,7 @@ if core.is_compiled_with_mkldnn(): ...@@ -38,6 +38,7 @@ if core.is_compiled_with_mkldnn():
macros.append(("PADDLE_WITH_MKLDNN", None)) macros.append(("PADDLE_WITH_MKLDNN", None))
if core.is_compiled_with_nccl(): if core.is_compiled_with_nccl():
macros.append(("PADDLE_WITH_NCCL", None)) macros.append(("PADDLE_WITH_NCCL", None))
macros.append(("THRUST_IGNORE_CUB_VERSION_CHECK", None))
include_dirs = list(paddle_includes) + [cwd] include_dirs = list(paddle_includes) + [cwd]
setup(name=os.getenv("MODULE_NAME", "custom_raw_op_kernel_op_setup"), setup(name=os.getenv("MODULE_NAME", "custom_raw_op_kernel_op_setup"),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册