diff --git a/paddle/fluid/operators/load_op.cc b/paddle/fluid/operators/load_op.cc index 662a616543f23e55619a122f757cfca9e11df030..abf7becb2e7fc134e3a52ec4c118847c14a20b9c 100644 --- a/paddle/fluid/operators/load_op.cc +++ b/paddle/fluid/operators/load_op.cc @@ -12,7 +12,6 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. */ #include -#include #include "paddle/fluid/framework/data_type_transform.h" #include "paddle/fluid/framework/op_registry.h" @@ -53,9 +52,6 @@ class LoadOp : public framework::OperatorBase { auto in_dtype = framework::ToDataType(tensor->type()); auto out_dtype = load_as_fp16 ? framework::proto::VarType::FP16 : in_dtype; - std::cout << "In load op: " << std::endl; - std::cout << "before conversion block" << std::endl; - if (in_dtype != out_dtype) { // convert to float16 tensor auto in_kernel_type = framework::OpKernelType(in_dtype, place); @@ -63,20 +59,15 @@ class LoadOp : public framework::OperatorBase { framework::LoDTensor fp16_tensor; // copy LoD info to the new tensor fp16_tensor.set_lod(tensor->lod()); - std::cout << "before TransDataType" << std::endl; framework::TransDataType(in_kernel_type, out_kernel_type, *tensor, &fp16_tensor); - std::cout << "after TransDataType" << std::endl; + // reset output tensor out_var->Clear(); tensor = out_var->GetMutable(); tensor->set_lod(fp16_tensor.lod()); - std::cout << "before TransDataType" << std::endl; tensor->ShareDataWith(fp16_tensor); - std::cout << "after TransDataType" << std::endl; } - - std::cout << "Out of load op: " << std::endl; } }; diff --git a/paddle/fluid/operators/save_load_op_test.cc b/paddle/fluid/operators/save_load_op_test.cc index 9c15461305c45363151cd70fb3f9614e860660b1..8d5c17490c08075723a1180d9b16e6000fc5a779 100644 --- a/paddle/fluid/operators/save_load_op_test.cc +++ b/paddle/fluid/operators/save_load_op_test.cc @@ -127,17 +127,13 @@ TEST(LoadFP16Op, CPU) { auto load_var = scope.Var("out_var"); auto load_op = paddle::framework::OpRegistry::CreateOp( "load", {}, {{"Out", {"out_var"}}}, attrs); - LOG(INFO) << "before load op run"; load_op->Run(scope, place); - LOG(INFO) << "after load op run"; auto target = load_var->Get(); paddle::platform::float16* actual = target.data(); - LOG(INFO) << "after target->data"; for (int64_t i = 0; i < tensor->numel(); ++i) { EXPECT_EQ(expect[i], static_cast(actual[i])); } - LOG(INFO) << "after expect equal"; auto& actual_lod = target.lod(); EXPECT_EQ(expect_lod.size(), actual_lod.size());