From 719f74198b1693b60565138b820131e592ab1221 Mon Sep 17 00:00:00 2001 From: zhangkaihuo <412475406@qq.com> Date: Tue, 4 Jan 2022 14:01:00 +0800 Subject: [PATCH] Modify macro definition to support arm (#38642) --- paddle/fluid/platform/dynload/cusparse.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/platform/dynload/cusparse.h b/paddle/fluid/platform/dynload/cusparse.h index 9f53c06ac02..fc842a3377b 100644 --- a/paddle/fluid/platform/dynload/cusparse.h +++ b/paddle/fluid/platform/dynload/cusparse.h @@ -41,7 +41,7 @@ extern void *cusparse_dso_handle; }; \ extern DynLoad__##__name __name -#if !defined(PADDLE_WITH_ARM) +#if defined(PADDLE_WITH_CUDA) // The generic APIs is supported from CUDA10.1 #if CUDA_VERSION >= 10010 #define CUSPARSE_ROUTINE_EACH(__macro) \ -- GitLab