From ac1c0a40070856a3449ae6fac461e6e6f38453c5 Mon Sep 17 00:00:00 2001
From: zhangyang <zhangyang49@baidu.com>
Date: Tue, 21 Aug 2018 10:47:48 +0800
Subject: [PATCH] assign value forscale pointer

---
 src/operators/feed_op.h | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/operators/feed_op.h b/src/operators/feed_op.h
index 0ed3c448e2..33ccd574e3 100644
--- a/src/operators/feed_op.h
+++ b/src/operators/feed_op.h
@@ -49,6 +49,7 @@ class FeedOp : public framework::OperatorBase<DeviceType> {
     auto input_ptr = input->data<float>();
     Tensor *output = param_.Out();
     auto output_ptr = output->mutable_data<half>();
+    auto out_address = output->fpga_args().scale_pointer();
     fpga::BypassArgs args;
     args.convert_type = fpga::DATA_FP32_TO_FP16;
     args.layout_type = fpga::LAYOUT_CHW_TO_HWC;
@@ -59,6 +60,7 @@ class FeedOp : public framework::OperatorBase<DeviceType> {
     args.image.pad_height = 0;
     args.image.pad_width = 0;
     args.output.address = output_ptr;
+    args.output.scale_address = out_address;
     fpga::PerformBypass(args);
   }
 
-- 
GitLab