Modify macro definition to support arm (#38642)

719f7419 · zhangkaihuo · GitHub · 9f0958fa · 719f7419
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/platform/dynload/cusparse.h paddle/fluid/platform/dynload/cusparse.h +1 -1

未找到文件。
--- 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) \