fix: fix ci compile error when without gpu

PaddlePaddle · Nov 28, 2023 · c65442a · c65442a
1 parent a3daaa2
commit c65442a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/paddle/fluid/pir/transforms/fusion/fused_weight_only_linear_pass.cc b/paddle/fluid/pir/transforms/fusion/fused_weight_only_linear_pass.cc
@@ -23,8 +23,11 @@
 namespace {
 
 inline int getSMVersion() {
-  int sm_version = paddle::platform::GetGPUComputeCapability(
+  int sm_version = 80;
+#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
+  sm_version = paddle::platform::GetGPUComputeCapability(
       paddle::platform::GetCurrentDeviceId());
+#endif
   return sm_version;
 }