未验证 提交 719f7419 编写于 作者: Z zhangkaihuo 提交者: GitHub

Modify macro definition to support arm (#38642)

上级 9f0958fa
...@@ -41,7 +41,7 @@ extern void *cusparse_dso_handle; ...@@ -41,7 +41,7 @@ extern void *cusparse_dso_handle;
}; \ }; \
extern DynLoad__##__name __name extern DynLoad__##__name __name
#if !defined(PADDLE_WITH_ARM) #if defined(PADDLE_WITH_CUDA)
// The generic APIs is supported from CUDA10.1 // The generic APIs is supported from CUDA10.1
#if CUDA_VERSION >= 10010 #if CUDA_VERSION >= 10010
#define CUSPARSE_ROUTINE_EACH(__macro) \ #define CUSPARSE_ROUTINE_EACH(__macro) \
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册