diff --git a/paddle/fluid/imperative/layer.h b/paddle/fluid/imperative/layer.h index 0db292992a29a87590667ce0652ac344014efea1..855e3e81995c0cb9a6af1c687d9769332e1e53d2 100644 --- a/paddle/fluid/imperative/layer.h +++ b/paddle/fluid/imperative/layer.h @@ -568,7 +568,7 @@ class OpBase { void CheckAttrs() { auto& info = op_->Info(); if (info.Checker() != nullptr) { - info.Checker()->Check(&attrs_); + info.Checker()->Check(&attrs_, true); } } diff --git a/paddle/fluid/imperative/tracer.cc b/paddle/fluid/imperative/tracer.cc index 0c8a4983ef420a26d595d34ee94b7d9d5c32b3f1..e99bd6700a1b8f9f446517d80e76a51a1bdb8b3f 100644 --- a/paddle/fluid/imperative/tracer.cc +++ b/paddle/fluid/imperative/tracer.cc @@ -92,7 +92,7 @@ void Tracer::TraceOp(const std::string& type, const NameVarBaseMap& ins, const auto& op_info = op->Info(); auto* attr_checker = op_info.Checker(); if (attr_checker) { - attr_checker->Check(&attrs); + attr_checker->Check(&attrs, true); } OpBase::Run(*op, ins, outs, attrs, place);