fix compile

PaddlePaddle · Sep 25, 2023 · 5048640 · 5048640
1 parent 2bac6f6
commit 5048640
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/paddle/fluid/primitive/codegen/templates/rule/vjp/generated/generated_vjp.cc.j2 b/paddle/fluid/primitive/codegen/templates/rule/vjp/generated/generated_vjp.cc.j2
@@ -102,6 +102,7 @@ vjp_res = ConstructVjpResultByStopGradients(vjp_res, stop_gradients);
 
 {% macro body_prim(api) %}
 FLAGS_tensor_operants_mode = "static";
+VLOG(4) << "PIR Static Prim API {{api.name}} call";
   {% for i in range(api.outputs|length) %}
     {% if api.outputs[i].typename=='Tensor' %}
 paddle::Tensor* {{api.outputs[i].name}} = !stop_gradients[{{i}}][0] ? &vjp_res[{{i}}][0] : nullptr; 

diff --git a/paddle/fluid/primitive/utils/CMakeLists.txt b/paddle/fluid/primitive/utils/CMakeLists.txt
@@ -7,4 +7,4 @@ endif()
 cc_library(
   primitive_static_utils_experimental
   SRCS static_utils.cc
-  DEPS phi common_infer_shape_functions)
+  DEPS phi common_infer_shape_functions pd_op_dialect_api)