fix cuda error (#5509)

PaddlePaddle · Feb 23, 2021 · 7763db2 · 7763db2
1 parent a759615
commit 7763db2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lite/core/program.cc b/lite/core/program.cc
@@ -398,7 +398,7 @@ void Program::PrepareWorkspace(
       VLOG(4) << "Var " << var_name << " in block " << block_idx;
       VLOG(4) << " - type " << static_cast<int>(var_type);
 
-#ifdef LITE_WITH_XPU
+#if defined(LITE_WITH_XPU) || defined(LITE_WITH_CUDA)
       if (!var_desc->Persistable()) {
 #endif
         // Collect precision info into var_type_map_
@@ -414,7 +414,7 @@ void Program::PrepareWorkspace(
           var_type_map_[var_name] = LiteType::GetTensorListTy(
               TARGET(kUnk), PRECISION(kUnk), DATALAYOUT(kUnk));
         }
-#ifdef LITE_WITH_XPU
+#if defined(LITE_WITH_XPU) || defined(LITE_WITH_CUDA)
       }
 #endif