Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
Paddle-Lite
提交
c4cce985
P
Paddle-Lite
项目概览
PaddlePaddle
/
Paddle-Lite
通知
331
Star
4
Fork
1
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
271
列表
看板
标记
里程碑
合并请求
78
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle-Lite
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
271
Issue
271
列表
看板
标记
里程碑
合并请求
78
合并请求
78
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
c4cce985
编写于
10月 20, 2018
作者:
Z
zhangyang
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
change CMakelist for FPGA track
上级
5c289362
变更
19
隐藏空白更改
内联
并排
Showing
19 changed file
with
108 addition
and
208 deletion
+108
-208
CMakeLists.txt
CMakeLists.txt
+11
-5
src/common/variant.h
src/common/variant.h
+1
-1
src/fpga/api.cpp
src/fpga/api.cpp
+8
-8
src/fpga/api.h
src/fpga/api.h
+14
-30
src/fpga/filter.cpp
src/fpga/filter.cpp
+4
-1
src/operators/kernel/fpga/conv_add_bn_kernel.cpp
src/operators/kernel/fpga/conv_add_bn_kernel.cpp
+5
-4
src/operators/kernel/fpga/conv_add_bn_relu_kernel.cpp
src/operators/kernel/fpga/conv_add_bn_relu_kernel.cpp
+5
-4
src/operators/kernel/fpga/conv_add_relu_kernel.cpp
src/operators/kernel/fpga/conv_add_relu_kernel.cpp
+5
-4
src/operators/kernel/fpga/conv_bn_kernel.cpp
src/operators/kernel/fpga/conv_bn_kernel.cpp
+5
-4
src/operators/kernel/fpga/conv_bn_relu_kernel.cpp
src/operators/kernel/fpga/conv_bn_relu_kernel.cpp
+5
-4
src/operators/kernel/fpga/fc_relu_kernel.cpp
src/operators/kernel/fpga/fc_relu_kernel.cpp
+3
-3
src/operators/kernel/fpga/fusion_fc_kernel.cpp
src/operators/kernel/fpga/fusion_fc_kernel.cpp
+3
-3
src/operators/kernel/fpga/mul_kernel.cpp
src/operators/kernel/fpga/mul_kernel.cpp
+0
-71
src/operators/math/gemm_int8.cpp
src/operators/math/gemm_int8.cpp
+2
-2
src/operators/op_param.h
src/operators/op_param.h
+30
-30
src/operators/sum_op.cpp
src/operators/sum_op.cpp
+0
-1
test/CMakeLists.txt
test/CMakeLists.txt
+2
-29
test/fpga/test_resnet50.cpp
test/fpga/test_resnet50.cpp
+5
-1
tools/op.cmake
tools/op.cmake
+0
-3
未找到文件。
CMakeLists.txt
浏览文件 @
c4cce985
...
@@ -2,9 +2,9 @@ cmake_minimum_required(VERSION 3.0)
...
@@ -2,9 +2,9 @@ cmake_minimum_required(VERSION 3.0)
project
(
paddle-mobile
)
project
(
paddle-mobile
)
# select the platform to build
# select the platform to build
option
(
CPU
"armv7 with neon support"
O
N
)
option
(
CPU
"armv7 with neon support"
O
FF
)
option
(
MALI_GPU
"mali gpu support"
OFF
)
option
(
MALI_GPU
"mali gpu support"
OFF
)
option
(
FPGA
"fpga support"
O
FF
)
option
(
FPGA
"fpga support"
O
N
)
option
(
USE_OPENMP
"openmp support"
OFF
)
option
(
USE_OPENMP
"openmp support"
OFF
)
option
(
DEBUGING
"enable debug mode"
ON
)
option
(
DEBUGING
"enable debug mode"
ON
)
...
@@ -29,7 +29,10 @@ if(DEBUGING)
...
@@ -29,7 +29,10 @@ if(DEBUGING)
message
(
STATUS
"debugging mode"
)
message
(
STATUS
"debugging mode"
)
add_definitions
(
-DPADDLE_MOBILE_DEBUG
)
add_definitions
(
-DPADDLE_MOBILE_DEBUG
)
else
()
else
()
add_definitions
(
-fvisibility=hidden -fvisibility-inlines-hidden
)
if
(
FPGA
)
else
()
add_definitions
(
-fvisibility=hidden -fvisibility-inlines-hidden
)
endif
()
endif
()
endif
()
if
(
USE_EXCEPTION
)
if
(
USE_EXCEPTION
)
...
@@ -93,8 +96,7 @@ else()
...
@@ -93,8 +96,7 @@ else()
endif
()
endif
()
if
(
FPGA
)
if
(
FPGA
)
set
(
DEBUGING ON
)
message
(
"FPGA mode enabled"
)
add_definitions
(
-DPADDLE_MOBILE_DEBUG
)
add_definitions
(
-DPADDLE_MOBILE_FPGA
)
add_definitions
(
-DPADDLE_MOBILE_FPGA
)
else
()
else
()
file
(
GLOB_RECURSE _tmp_list src/operators/kernel/fpga/*.cpp src/operators/kernel/fpga/*.cc
)
file
(
GLOB_RECURSE _tmp_list src/operators/kernel/fpga/*.cpp src/operators/kernel/fpga/*.cc
)
...
@@ -177,6 +179,10 @@ if(DEBUGING)
...
@@ -177,6 +179,10 @@ if(DEBUGING)
else
()
else
()
add_subdirectory
(
test
)
add_subdirectory
(
test
)
endif
()
endif
()
elseif
(
FPGA
)
add_subdirectory
(
test
)
endif
()
endif
()
src/common/variant.h
浏览文件 @
c4cce985
...
@@ -13,7 +13,7 @@ See the License for the specific language governing permissions and
...
@@ -13,7 +13,7 @@ See the License for the specific language governing permissions and
limitations under the License. */
limitations under the License. */
#include <cstdlib>
#include <cstdlib>
#include <cstring>
#include "common/enforce.h"
#include "common/enforce.h"
#include "common/log.h"
#include "common/log.h"
...
...
src/fpga/api.cpp
浏览文件 @
c4cce985
...
@@ -22,7 +22,7 @@ limitations under the License. */
...
@@ -22,7 +22,7 @@ limitations under the License. */
#include "fpga/filter.h"
#include "fpga/filter.h"
#include "fpga/image.h"
#include "fpga/image.h"
#define FPGA_TEST_MODE
#define FPGA_TEST_MODE
//#define PADDLE_MOBILE_OS_LINUX
//
#define PADDLE_MOBILE_OS_LINUX
namespace
paddle_mobile
{
namespace
paddle_mobile
{
namespace
fpga
{
namespace
fpga
{
...
@@ -149,7 +149,7 @@ int ComputeBasicConv(const struct ConvArgs &args) {
...
@@ -149,7 +149,7 @@ int ComputeBasicConv(const struct ConvArgs &args) {
return
do_ioctl
(
IOCTL_CONFIG_CONV
,
&
args
);
return
do_ioctl
(
IOCTL_CONFIG_CONV
,
&
args
);
}
}
int
ComputeFpgaConv
(
const
struct
Wrapper
ConvArgs
&
args
)
{
int
ComputeFpgaConv
(
const
struct
Split
ConvArgs
&
args
)
{
#ifdef FPGA_TEST_MODE
#ifdef FPGA_TEST_MODE
DLOG
<<
"=============ComputeFPGAConv==========="
;
DLOG
<<
"=============ComputeFPGAConv==========="
;
DLOG
<<
" filter_num:"
<<
args
.
filter_num
DLOG
<<
" filter_num:"
<<
args
.
filter_num
...
@@ -194,8 +194,8 @@ int ComputeFpgaEWAdd(const struct EWAddArgs &args) {
...
@@ -194,8 +194,8 @@ int ComputeFpgaEWAdd(const struct EWAddArgs &args) {
#ifdef FPGA_TEST_MODE
#ifdef FPGA_TEST_MODE
DLOG
<<
"=============ComputeFpgaEWAdd==========="
;
DLOG
<<
"=============ComputeFpgaEWAdd==========="
;
DLOG
<<
" relu_enabled:"
<<
args
.
relu_enabled
DLOG
<<
" relu_enabled:"
<<
args
.
relu_enabled
<<
" const0:"
<<
fp16_2_fp32
(
shor
t
(
args
.
const0
))
<<
" const0:"
<<
fp16_2_fp32
(
int16_
t
(
args
.
const0
))
<<
" const1:"
<<
fp16_2_fp32
(
shor
t
(
args
.
const1
));
<<
" const1:"
<<
fp16_2_fp32
(
int16_
t
(
args
.
const1
));
DLOG
<<
" image0_address:"
<<
args
.
image0
.
address
DLOG
<<
" image0_address:"
<<
args
.
image0
.
address
<<
" image0_scale_address:"
<<
args
.
image0
.
scale_address
<<
" image0_scale_address:"
<<
args
.
image0
.
scale_address
<<
" image0_channels:"
<<
args
.
image0
.
channels
<<
" image0_channels:"
<<
args
.
image0
.
channels
...
@@ -383,10 +383,10 @@ void format_concat_output(framework::Tensor *out, int height, int width,
...
@@ -383,10 +383,10 @@ void format_concat_output(framework::Tensor *out, int height, int width,
out
->
reset_data_ptr
(
data_ptr
);
out
->
reset_data_ptr
(
data_ptr
);
}
}
void
fill_
conv_arg
(
struct
Wrapper
ConvArgs
*
arg
,
framework
::
Tensor
*
input
,
void
fill_
split_arg
(
struct
Split
ConvArgs
*
arg
,
framework
::
Tensor
*
input
,
framework
::
Tensor
*
out
,
framework
::
Tensor
*
filter
,
framework
::
Tensor
*
out
,
framework
::
Tensor
*
filter
,
bool
relu_enabled
,
int
group_num
,
int
stride_h
,
int
stride_w
,
bool
relu_enabled
,
int
group_num
,
int
stride_h
,
int
padding_h
,
int
padding_w
,
float
*
bs_ptr
)
{
int
stride_w
,
int
padding_h
,
int
padding_w
,
float
*
bs_ptr
)
{
auto
input_ptr
=
input
->
data
<
float
>
();
auto
input_ptr
=
input
->
data
<
float
>
();
auto
filter_ptr
=
filter
->
data
<
float
>
();
auto
filter_ptr
=
filter
->
data
<
float
>
();
auto
out_ptr
=
out
->
data
<
float
>
();
auto
out_ptr
=
out
->
data
<
float
>
();
...
...
src/fpga/api.h
浏览文件 @
c4cce985
...
@@ -89,7 +89,7 @@ struct ConcatArgs {
...
@@ -89,7 +89,7 @@ struct ConcatArgs {
uint32_t
width
;
uint32_t
width
;
};
};
struct
Wrapper
ConvArgs
{
struct
Split
ConvArgs
{
uint32_t
split_num
;
uint32_t
split_num
;
uint32_t
group_num
;
uint32_t
group_num
;
uint32_t
filter_num
;
uint32_t
filter_num
;
...
@@ -98,6 +98,14 @@ struct WrapperConvArgs {
...
@@ -98,6 +98,14 @@ struct WrapperConvArgs {
struct
ConcatArgs
concat_arg
;
struct
ConcatArgs
concat_arg
;
};
};
struct
GroupConvArgs
{
uint32_t
group_num
;
uint32_t
filter_num
;
struct
ImageOutputArgs
output
;
struct
SplitConvArgs
*
conv_args
;
struct
ConcatArgs
concat_arg
;
};
struct
PoolingArgs
{
struct
PoolingArgs
{
int16_t
mode
;
// mode: 0:max, 1:avg
int16_t
mode
;
// mode: 0:max, 1:avg
half
kernel_reciprocal
;
half
kernel_reciprocal
;
...
@@ -159,30 +167,6 @@ struct MemoryCacheArgs {
...
@@ -159,30 +167,6 @@ struct MemoryCacheArgs {
#define IOCTL_FPGA_REG_READ _IOW(IOCTL_FPGA_MAGIC, 28, struct FpgaRegReadArgs)
#define IOCTL_FPGA_REG_READ _IOW(IOCTL_FPGA_MAGIC, 28, struct FpgaRegReadArgs)
#define IOCTL_FPGA_REG_WRITE _IOW(IOCTL_FPGA_MAGIC, 29, struct FpgaRegWriteArgs)
#define IOCTL_FPGA_REG_WRITE _IOW(IOCTL_FPGA_MAGIC, 29, struct FpgaRegWriteArgs)
enum
FPGA_ERR_TYPE
{
ERR_IOCTL_CMD
=
-
1
,
ERR_TIMEOUT
=
-
2
,
ERR_COMPLETION_TIMEOUT
=
-
3
,
ERR_INVALID_FPGA_ADDR
=
-
4
,
ERR_NOMEM
=
-
5
,
ERR_NO_RESERVE_MEM
=
-
6
,
ERR_COPY_FROM_USER
=
-
7
,
ERR_COPY_TO_USER
=
-
8
,
ERR_DEL_TIMER
=
-
9
,
ERR_ENABLE_MSI
=
-
10
,
ERR_REGISTER_IRQ
=
-
11
,
ERR_PCIE_REGISTER
=
-
12
,
ERR_PCIE_PROBE
=
-
13
,
ERR_REGISTER_BLOCK
=
-
14
,
ERR_ALLOC_GENDISK
=
-
15
,
ERR_INIT_QUEUE
=
-
16
,
ERR_WAIT
=
-
17
,
ERR_ECC_ERROR
=
-
31
,
ERR_FPGA_FAIL_STOP
=
-
64
,
ERR_FPGA_DEBUG_STOP
=
-
113
,
DEV_TMP_UNAVAILABLE
=
-
128
};
//============================== API =============================
//============================== API =============================
int
open_device
();
int
open_device
();
...
@@ -195,7 +179,7 @@ int fpga_flush(void* address, size_t size);
...
@@ -195,7 +179,7 @@ int fpga_flush(void* address, size_t size);
int
fpga_invalidate
(
void
*
address
,
size_t
size
);
int
fpga_invalidate
(
void
*
address
,
size_t
size
);
int
PerformBypass
(
const
struct
BypassArgs
&
args
);
int
PerformBypass
(
const
struct
BypassArgs
&
args
);
int
ComputeFpgaConv
(
const
struct
Wrapper
ConvArgs
&
args
);
int
ComputeFpgaConv
(
const
struct
Split
ConvArgs
&
args
);
int
ComputeFpgaPool
(
const
struct
PoolingArgs
&
args
);
int
ComputeFpgaPool
(
const
struct
PoolingArgs
&
args
);
int
ComputeFpgaEWAdd
(
const
struct
EWAddArgs
&
args
);
int
ComputeFpgaEWAdd
(
const
struct
EWAddArgs
&
args
);
int
ComputeFPGAConcat
(
const
struct
ConcatArgs
&
args
);
int
ComputeFPGAConcat
(
const
struct
ConcatArgs
&
args
);
...
@@ -220,10 +204,10 @@ void format_bias_scale_array(float** bias_scale_array,
...
@@ -220,10 +204,10 @@ void format_bias_scale_array(float** bias_scale_array,
void
format_concat_output
(
framework
::
Tensor
*
out
,
int
height
,
int
width
,
void
format_concat_output
(
framework
::
Tensor
*
out
,
int
height
,
int
width
,
int
image_num
,
uint32_t
*
channel_num
);
int
image_num
,
uint32_t
*
channel_num
);
void
fill_
conv_arg
(
struct
Wrapper
ConvArgs
*
arg
,
framework
::
Tensor
*
input
,
void
fill_
split_arg
(
struct
Split
ConvArgs
*
arg
,
framework
::
Tensor
*
input
,
framework
::
Tensor
*
out
,
framework
::
Tensor
*
filter
,
framework
::
Tensor
*
out
,
framework
::
Tensor
*
filter
,
bool
relu_enabled
,
int
group_num
,
int
stride_h
,
int
stride_w
,
bool
relu_enabled
,
int
group_num
,
int
stride_h
,
int
padding_h
,
int
padding_w
,
float
*
bs_ptr
);
int
stride_w
,
int
padding_h
,
int
padding_w
,
float
*
bs_ptr
);
half
fp32_2_fp16
(
float
fp32_num
);
half
fp32_2_fp16
(
float
fp32_num
);
float
fp16_2_fp32
(
half
fp16_num
);
float
fp16_2_fp32
(
half
fp16_num
);
...
...
src/fpga/filter.cpp
浏览文件 @
c4cce985
...
@@ -21,7 +21,10 @@ namespace paddle_mobile {
...
@@ -21,7 +21,10 @@ namespace paddle_mobile {
namespace
fpga
{
namespace
fpga
{
namespace
filter
{
namespace
filter
{
int
calc_division_capacity
(
int
chw
)
{
return
2048
/
((
chw
+
15
)
/
16
)
*
32
;
}
int
calc_division_capacity
(
int
chw
)
{
int
n
=
2048
/
((
chw
+
15
)
/
16
)
*
32
;
return
n
<
2048
?
n
:
2048
;
}
int
calc_split_num
(
int
num
,
int
division_capacity
)
{
int
calc_split_num
(
int
num
,
int
division_capacity
)
{
return
(
num
+
division_capacity
-
1
)
/
division_capacity
;
return
(
num
+
division_capacity
-
1
)
/
division_capacity
;
...
...
src/operators/kernel/fpga/conv_add_bn_kernel.cpp
浏览文件 @
c4cce985
...
@@ -66,10 +66,11 @@ bool ConvAddBNKernel<FPGA, float>::Init(FusionConvAddBNParam<FPGA> *param) {
...
@@ -66,10 +66,11 @@ bool ConvAddBNKernel<FPGA, float>::Init(FusionConvAddBNParam<FPGA> *param) {
fpga
::
format_bias_scale_array
(
&
bs_ptr
,
element_num_per_div
,
channel
);
fpga
::
format_bias_scale_array
(
&
bs_ptr
,
element_num_per_div
,
channel
);
fpga
::
format_fp16_ofm
(
out
);
fpga
::
format_fp16_ofm
(
out
);
fpga
::
WrapperConvArgs
conv_arg
=
{
0
};
fpga
::
SplitConvArgs
conv_arg
=
{
0
};
fpga
::
fill_conv_arg
(
&
conv_arg
,
input
,
out
,
filter
,
relu_enabled
,
fpga
::
fill_split_arg
(
&
conv_arg
,
input
,
out
,
filter
,
relu_enabled
,
param
->
Groups
(),
param
->
Strides
()[
0
],
param
->
Strides
()[
1
],
param
->
Groups
(),
param
->
Strides
()[
0
],
param
->
Paddings
()[
0
],
param
->
Paddings
()[
1
],
bs_ptr
);
param
->
Strides
()[
1
],
param
->
Paddings
()[
0
],
param
->
Paddings
()[
1
],
bs_ptr
);
param
->
SetFpgaArgs
(
conv_arg
);
param
->
SetFpgaArgs
(
conv_arg
);
return
true
;
return
true
;
...
...
src/operators/kernel/fpga/conv_add_bn_relu_kernel.cpp
浏览文件 @
c4cce985
...
@@ -65,10 +65,11 @@ bool ConvAddBNReluKernel<FPGA, float>::Init(
...
@@ -65,10 +65,11 @@ bool ConvAddBNReluKernel<FPGA, float>::Init(
fpga
::
format_fp16_ofm
(
out
);
fpga
::
format_fp16_ofm
(
out
);
fpga
::
WrapperConvArgs
conv_arg
=
{
0
};
fpga
::
SplitConvArgs
conv_arg
=
{
0
};
fpga
::
fill_conv_arg
(
&
conv_arg
,
input
,
out
,
filter
,
relu_enabled
,
fpga
::
fill_split_arg
(
&
conv_arg
,
input
,
out
,
filter
,
relu_enabled
,
param
->
Groups
(),
param
->
Strides
()[
0
],
param
->
Strides
()[
1
],
param
->
Groups
(),
param
->
Strides
()[
0
],
param
->
Paddings
()[
0
],
param
->
Paddings
()[
1
],
bs_ptr
);
param
->
Strides
()[
1
],
param
->
Paddings
()[
0
],
param
->
Paddings
()[
1
],
bs_ptr
);
param
->
SetFpgaArgs
(
conv_arg
);
param
->
SetFpgaArgs
(
conv_arg
);
return
true
;
return
true
;
}
}
...
...
src/operators/kernel/fpga/conv_add_relu_kernel.cpp
浏览文件 @
c4cce985
...
@@ -47,10 +47,11 @@ bool ConvAddReluKernel<FPGA, float>::Init(FusionConvAddReluParam<FPGA> *param) {
...
@@ -47,10 +47,11 @@ bool ConvAddReluKernel<FPGA, float>::Init(FusionConvAddReluParam<FPGA> *param) {
fpga
::
format_fp16_ofm
(
out
);
fpga
::
format_fp16_ofm
(
out
);
fpga
::
WrapperConvArgs
conv_arg
=
{
0
};
fpga
::
SplitConvArgs
conv_arg
=
{
0
};
fpga
::
fill_conv_arg
(
&
conv_arg
,
input
,
out
,
filter
,
relu_enabled
,
fpga
::
fill_split_arg
(
&
conv_arg
,
input
,
out
,
filter
,
relu_enabled
,
param
->
Groups
(),
param
->
Strides
()[
0
],
param
->
Strides
()[
1
],
param
->
Groups
(),
param
->
Strides
()[
0
],
param
->
Paddings
()[
0
],
param
->
Paddings
()[
1
],
bs_ptr
);
param
->
Strides
()[
1
],
param
->
Paddings
()[
0
],
param
->
Paddings
()[
1
],
bs_ptr
);
param
->
SetFpgaArgs
(
conv_arg
);
param
->
SetFpgaArgs
(
conv_arg
);
return
true
;
return
true
;
}
}
...
...
src/operators/kernel/fpga/conv_bn_kernel.cpp
浏览文件 @
c4cce985
...
@@ -59,10 +59,11 @@ bool ConvBNKernel<FPGA, float>::Init(FusionConvBNParam<FPGA> *param) {
...
@@ -59,10 +59,11 @@ bool ConvBNKernel<FPGA, float>::Init(FusionConvBNParam<FPGA> *param) {
fpga
::
format_fp16_ofm
(
out
);
fpga
::
format_fp16_ofm
(
out
);
fpga
::
WrapperConvArgs
conv_arg
=
{
0
};
fpga
::
SplitConvArgs
conv_arg
=
{
0
};
fpga
::
fill_conv_arg
(
&
conv_arg
,
input
,
out
,
filter
,
relu_enabled
,
fpga
::
fill_split_arg
(
&
conv_arg
,
input
,
out
,
filter
,
relu_enabled
,
param
->
Groups
(),
param
->
Strides
()[
0
],
param
->
Strides
()[
1
],
param
->
Groups
(),
param
->
Strides
()[
0
],
param
->
Paddings
()[
0
],
param
->
Paddings
()[
1
],
bs_ptr
);
param
->
Strides
()[
1
],
param
->
Paddings
()[
0
],
param
->
Paddings
()[
1
],
bs_ptr
);
param
->
SetFpgaArgs
(
conv_arg
);
param
->
SetFpgaArgs
(
conv_arg
);
return
true
;
return
true
;
}
}
...
...
src/operators/kernel/fpga/conv_bn_relu_kernel.cpp
浏览文件 @
c4cce985
...
@@ -59,10 +59,11 @@ bool ConvBNReluKernel<FPGA, float>::Init(FusionConvBNReluParam<FPGA> *param) {
...
@@ -59,10 +59,11 @@ bool ConvBNReluKernel<FPGA, float>::Init(FusionConvBNReluParam<FPGA> *param) {
fpga
::
format_fp16_ofm
(
out
);
fpga
::
format_fp16_ofm
(
out
);
fpga
::
WrapperConvArgs
conv_arg
=
{
0
};
fpga
::
SplitConvArgs
conv_arg
=
{
0
};
fpga
::
fill_conv_arg
(
&
conv_arg
,
input
,
out
,
filter
,
relu_enabled
,
fpga
::
fill_split_arg
(
&
conv_arg
,
input
,
out
,
filter
,
relu_enabled
,
param
->
Groups
(),
param
->
Strides
()[
0
],
param
->
Strides
()[
1
],
param
->
Groups
(),
param
->
Strides
()[
0
],
param
->
Paddings
()[
0
],
param
->
Paddings
()[
1
],
bs_ptr
);
param
->
Strides
()[
1
],
param
->
Paddings
()[
0
],
param
->
Paddings
()[
1
],
bs_ptr
);
param
->
SetFpgaArgs
(
conv_arg
);
param
->
SetFpgaArgs
(
conv_arg
);
return
true
;
return
true
;
}
}
...
...
src/operators/kernel/fpga/fc_relu_kernel.cpp
浏览文件 @
c4cce985
...
@@ -53,9 +53,9 @@ bool FusionFcReluKernel<FPGA, float>::Init(FusionFcReluParam<FPGA> *param) {
...
@@ -53,9 +53,9 @@ bool FusionFcReluKernel<FPGA, float>::Init(FusionFcReluParam<FPGA> *param) {
fpga
::
format_bias_scale_array
(
&
bs_ptr
,
element_num_per_div
,
channel
);
fpga
::
format_bias_scale_array
(
&
bs_ptr
,
element_num_per_div
,
channel
);
fpga
::
format_fp16_ofm
(
out
);
fpga
::
format_fp16_ofm
(
out
);
fpga
::
Wrapper
ConvArgs
conv_arg
=
{
0
};
fpga
::
Split
ConvArgs
conv_arg
=
{
0
};
fpga
::
fill_
conv_arg
(
&
conv_arg
,
input_x
,
out
,
filter
,
relu_enabled
,
1
,
1
,
1
,
0
,
fpga
::
fill_
split_arg
(
&
conv_arg
,
input_x
,
out
,
filter
,
relu_enabled
,
1
,
1
,
1
,
0
,
bs_ptr
);
0
,
0
,
bs_ptr
);
param
->
SetFpgaArgs
(
conv_arg
);
param
->
SetFpgaArgs
(
conv_arg
);
return
true
;
return
true
;
}
}
...
...
src/operators/kernel/fpga/fusion_fc_kernel.cpp
浏览文件 @
c4cce985
...
@@ -54,9 +54,9 @@ bool FusionFcKernel<FPGA, float>::Init(FusionFcParam<FPGA> *param) {
...
@@ -54,9 +54,9 @@ bool FusionFcKernel<FPGA, float>::Init(FusionFcParam<FPGA> *param) {
fpga
::
format_bias_scale_array
(
&
bs_ptr
,
element_num_per_div
,
channel
);
fpga
::
format_bias_scale_array
(
&
bs_ptr
,
element_num_per_div
,
channel
);
fpga
::
format_fp16_ofm
(
out
);
fpga
::
format_fp16_ofm
(
out
);
fpga
::
Wrapper
ConvArgs
conv_arg
=
{
0
};
fpga
::
Split
ConvArgs
conv_arg
=
{
0
};
fpga
::
fill_
conv_arg
(
&
conv_arg
,
input_x
,
out
,
filter
,
relu_enabled
,
1
,
1
,
1
,
0
,
fpga
::
fill_
split_arg
(
&
conv_arg
,
input_x
,
out
,
filter
,
relu_enabled
,
1
,
1
,
1
,
0
,
bs_ptr
);
0
,
0
,
bs_ptr
);
param
->
SetFpgaArgs
(
conv_arg
);
param
->
SetFpgaArgs
(
conv_arg
);
return
true
;
return
true
;
}
}
...
...
src/operators/kernel/fpga/mul_kernel.cpp
已删除
100644 → 0
浏览文件 @
5c289362
/* Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
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. */
#ifdef MUL_OP
#include "operators/kernel/mul_kernel.h"
namespace
paddle_mobile
{
namespace
operators
{
template
<
>
bool
MulKernel
<
FPGA
,
float
>::
Init
(
MulParam
<
FPGA
>
*
param
)
{
bool
relu_enabled
=
false
;
auto
input_x
=
const_cast
<
LoDTensor
*>
(
param
->
InputX
());
auto
filter
=
const_cast
<
LoDTensor
*>
(
param
->
InputY
());
auto
out
=
param
->
Out
();
PADDLE_MOBILE_ENFORCE
(
input_x
->
dims
()[
1
]
==
filter
->
dims
()[
0
],
"Image channel should be equal to weight number"
);
int
channel
=
(
uint32_t
)
out
->
dims
()[
1
];
auto
bs_ptr
=
(
float
*
)
fpga
::
fpga_malloc
(
2
*
channel
*
sizeof
(
float
));
// NOLINT
for
(
int
i
=
0
;
i
<
channel
;
i
++
)
{
bs_ptr
[
i
+
channel
]
=
1
;
bs_ptr
[
i
]
=
0
;
}
int
num
=
(
uint32_t
)
filter
->
dims
()[
1
];
int
chw
=
(
uint32_t
)
filter
->
dims
()[
0
];
PADDLE_MOBILE_ENFORCE
(
chw
==
input_x
->
numel
(),
"Filter element num should be equal to IFM element num"
);
int
height
=
(
uint32_t
)
input_x
->
dims
()[
2
];
int
width
=
(
uint32_t
)
input_x
->
dims
()[
3
];
int
filter_channel
=
chw
/
height
/
width
;
out
->
Resize
(
framework
::
make_ddim
({
1
,
channel
,
1
,
1
}));
filter
->
Resize
(
framework
::
make_ddim
({
num
,
filter_channel
,
height
,
width
}));
float
max_value
=
fpga
::
filter_find_max
(
filter
);
fpga
::
format_fc_filter
(
filter
,
max_value
);
int
element_num_per_div
=
fpga
::
get_filter_num_per_div
(
filter
,
1
);
fpga
::
format_bias_scale_array
(
&
bs_ptr
,
element_num_per_div
,
channel
);
fpga
::
format_fp16_ofm
(
out
);
fpga
::
WrapperConvArgs
conv_arg
=
{
0
};
fpga
::
fill_conv_arg
(
&
conv_arg
,
input_x
,
out
,
filter
,
relu_enabled
,
1
,
1
,
1
,
0
,
0
,
bs_ptr
);
param
->
SetFpgaArgs
(
conv_arg
);
return
true
;
}
template
<
>
void
MulKernel
<
FPGA
,
float
>::
Compute
(
const
MulParam
<
FPGA
>
&
param
)
const
{
fpga
::
ComputeFpgaConv
(
param
.
FpgaArgs
());
}
}
// namespace operators
}
// namespace paddle_mobile
#endif
src/operators/math/gemm_int8.cpp
浏览文件 @
c4cce985
...
@@ -652,7 +652,7 @@ void Gemm::WriteBasic(int32_t mc, int32_t nc, int32_t *c, int32_t *C,
...
@@ -652,7 +652,7 @@ void Gemm::WriteBasic(int32_t mc, int32_t nc, int32_t *c, int32_t *C,
int32_t
*
C0
,
*
c0
;
int32_t
*
C0
,
*
c0
;
c_ptr
=
c
;
c_ptr
=
c
;
C_ptr
=
C
;
C_ptr
=
C
;
if
(
nc1
>
0
)
{
/*
if (nc1 > 0) {
asm volatile(
asm volatile(
"subs %[mc], %[mc], #1 \n\t"
"subs %[mc], %[mc], #1 \n\t"
"blt end_mc_%= \n\t"
"blt end_mc_%= \n\t"
...
@@ -684,7 +684,7 @@ void Gemm::WriteBasic(int32_t mc, int32_t nc, int32_t *c, int32_t *C,
...
@@ -684,7 +684,7 @@ void Gemm::WriteBasic(int32_t mc, int32_t nc, int32_t *c, int32_t *C,
: [C_ptr] "r"(C_ptr), [c_ptr] "r"(c_ptr), [mc] "r"(m), [nc1] "r"(nc1),
: [C_ptr] "r"(C_ptr), [c_ptr] "r"(c_ptr), [mc] "r"(m), [nc1] "r"(nc1),
[step] "r"(step), [step1] "r"(step1)
[step] "r"(step), [step1] "r"(step1)
: "memory", "r5", "r6", "q0", "q1", "q2", "q3");
: "memory", "r5", "r6", "q0", "q1", "q2", "q3");
}
}
*/
if
(
_nc1
!=
0
)
{
if
(
_nc1
!=
0
)
{
for
(
int32_t
i
=
0
;
i
<
mc
;
i
++
)
{
for
(
int32_t
i
=
0
;
i
<
mc
;
i
++
)
{
...
...
src/operators/op_param.h
浏览文件 @
c4cce985
...
@@ -546,11 +546,11 @@ class MulParam : OpParam {
...
@@ -546,11 +546,11 @@ class MulParam : OpParam {
#ifdef PADDLE_MOBILE_FPGA
#ifdef PADDLE_MOBILE_FPGA
private:
private:
fpga
::
Wrapper
ConvArgs
fpga_conv_args
;
fpga
::
Split
ConvArgs
fpga_conv_args
;
public:
public:
const
fpga
::
Wrapper
ConvArgs
&
FpgaArgs
()
const
{
return
fpga_conv_args
;
}
const
fpga
::
Split
ConvArgs
&
FpgaArgs
()
const
{
return
fpga_conv_args
;
}
void
SetFpgaArgs
(
const
fpga
::
Wrapper
ConvArgs
&
args
)
{
fpga_conv_args
=
args
;
}
void
SetFpgaArgs
(
const
fpga
::
Split
ConvArgs
&
args
)
{
fpga_conv_args
=
args
;
}
#endif
#endif
};
};
#endif
#endif
...
@@ -1401,11 +1401,11 @@ class FusionFcParam : public OpParam {
...
@@ -1401,11 +1401,11 @@ class FusionFcParam : public OpParam {
#ifdef PADDLE_MOBILE_FPGA
#ifdef PADDLE_MOBILE_FPGA
private:
private:
fpga
::
Wrapper
ConvArgs
fpga_conv_args
;
fpga
::
Split
ConvArgs
fpga_conv_args
;
public:
public:
const
fpga
::
Wrapper
ConvArgs
&
FpgaArgs
()
const
{
return
fpga_conv_args
;
}
const
fpga
::
Split
ConvArgs
&
FpgaArgs
()
const
{
return
fpga_conv_args
;
}
void
SetFpgaArgs
(
const
fpga
::
Wrapper
ConvArgs
&
args
)
{
fpga_conv_args
=
args
;
}
void
SetFpgaArgs
(
const
fpga
::
Split
ConvArgs
&
args
)
{
fpga_conv_args
=
args
;
}
#endif
#endif
};
};
...
@@ -1441,11 +1441,11 @@ class FusionConvAddParam : public ConvParam<Dtype> {
...
@@ -1441,11 +1441,11 @@ class FusionConvAddParam : public ConvParam<Dtype> {
#ifdef PADDLE_MOBILE_FPGA
#ifdef PADDLE_MOBILE_FPGA
private:
private:
fpga
::
Wrapper
ConvArgs
fpga_conv_args
;
fpga
::
Split
ConvArgs
fpga_conv_args
;
public:
public:
const
fpga
::
Wrapper
ConvArgs
&
FpgaArgs
()
const
{
return
fpga_conv_args
;
}
const
fpga
::
Split
ConvArgs
&
FpgaArgs
()
const
{
return
fpga_conv_args
;
}
void
SetFpgaArgs
(
const
fpga
::
Wrapper
ConvArgs
&
args
)
{
fpga_conv_args
=
args
;
}
void
SetFpgaArgs
(
const
fpga
::
Split
ConvArgs
&
args
)
{
fpga_conv_args
=
args
;
}
#endif
#endif
};
};
...
@@ -1496,11 +1496,11 @@ class FusionConvAddPReluParam : public ConvParam<Dtype> {
...
@@ -1496,11 +1496,11 @@ class FusionConvAddPReluParam : public ConvParam<Dtype> {
#ifdef PADDLE_MOBILE_FPGA
#ifdef PADDLE_MOBILE_FPGA
private:
private:
fpga
::
Wrapper
ConvArgs
fpga_conv_args
;
fpga
::
Split
ConvArgs
fpga_conv_args
;
public:
public:
const
fpga
::
Wrapper
ConvArgs
&
FpgaArgs
()
const
{
return
fpga_conv_args
;
}
const
fpga
::
Split
ConvArgs
&
FpgaArgs
()
const
{
return
fpga_conv_args
;
}
void
SetFpgaArgs
(
const
fpga
::
Wrapper
ConvArgs
&
args
)
{
fpga_conv_args
=
args
;
}
void
SetFpgaArgs
(
const
fpga
::
Split
ConvArgs
&
args
)
{
fpga_conv_args
=
args
;
}
#endif
#endif
};
};
#endif
#endif
...
@@ -1554,11 +1554,11 @@ class FusionConvAddAddPReluParam : public ConvParam<Dtype> {
...
@@ -1554,11 +1554,11 @@ class FusionConvAddAddPReluParam : public ConvParam<Dtype> {
#ifdef PADDLE_MOBILE_FPGA
#ifdef PADDLE_MOBILE_FPGA
private:
private:
fpga
::
Wrapper
ConvArgs
fpga_conv_args
;
fpga
::
Split
ConvArgs
fpga_conv_args
;
public:
public:
const
fpga
::
Wrapper
ConvArgs
&
FpgaArgs
()
const
{
return
fpga_conv_args
;
}
const
fpga
::
Split
ConvArgs
&
FpgaArgs
()
const
{
return
fpga_conv_args
;
}
void
SetFpgaArgs
(
const
fpga
::
Wrapper
ConvArgs
&
args
)
{
fpga_conv_args
=
args
;
}
void
SetFpgaArgs
(
const
fpga
::
Split
ConvArgs
&
args
)
{
fpga_conv_args
=
args
;
}
#endif
#endif
};
};
#endif
#endif
...
@@ -1629,11 +1629,11 @@ class FusionConvAddBNReluParam : public ConvParam<Dtype> {
...
@@ -1629,11 +1629,11 @@ class FusionConvAddBNReluParam : public ConvParam<Dtype> {
#ifdef PADDLE_MOBILE_FPGA
#ifdef PADDLE_MOBILE_FPGA
private:
private:
fpga
::
Wrapper
ConvArgs
fpga_conv_args
;
fpga
::
Split
ConvArgs
fpga_conv_args
;
public:
public:
const
fpga
::
Wrapper
ConvArgs
&
FpgaArgs
()
const
{
return
fpga_conv_args
;
}
const
fpga
::
Split
ConvArgs
&
FpgaArgs
()
const
{
return
fpga_conv_args
;
}
void
SetFpgaArgs
(
const
fpga
::
Wrapper
ConvArgs
&
args
)
{
fpga_conv_args
=
args
;
}
void
SetFpgaArgs
(
const
fpga
::
Split
ConvArgs
&
args
)
{
fpga_conv_args
=
args
;
}
#endif
#endif
};
};
#endif
#endif
...
@@ -1715,11 +1715,11 @@ class FusionConvBNAddReluParam : public ConvParam<Dtype> {
...
@@ -1715,11 +1715,11 @@ class FusionConvBNAddReluParam : public ConvParam<Dtype> {
#ifdef PADDLE_MOBILE_FPGA
#ifdef PADDLE_MOBILE_FPGA
private:
private:
fpga
::
Wrapper
ConvArgs
fpga_conv_args
;
fpga
::
Split
ConvArgs
fpga_conv_args
;
public:
public:
const
fpga
::
Wrapper
ConvArgs
&
FpgaArgs
()
const
{
return
fpga_conv_args
;
}
const
fpga
::
Split
ConvArgs
&
FpgaArgs
()
const
{
return
fpga_conv_args
;
}
void
SetFpgaArgs
(
const
fpga
::
Wrapper
ConvArgs
&
args
)
{
fpga_conv_args
=
args
;
}
void
SetFpgaArgs
(
const
fpga
::
Split
ConvArgs
&
args
)
{
fpga_conv_args
=
args
;
}
#endif
#endif
};
};
#endif
#endif
...
@@ -1782,11 +1782,11 @@ class FusionConvBNParam : public ConvParam<Dtype> {
...
@@ -1782,11 +1782,11 @@ class FusionConvBNParam : public ConvParam<Dtype> {
#ifdef PADDLE_MOBILE_FPGA
#ifdef PADDLE_MOBILE_FPGA
private:
private:
fpga
::
Wrapper
ConvArgs
fpga_conv_args
;
fpga
::
Split
ConvArgs
fpga_conv_args
;
public:
public:
const
fpga
::
Wrapper
ConvArgs
&
FpgaArgs
()
const
{
return
fpga_conv_args
;
}
const
fpga
::
Split
ConvArgs
&
FpgaArgs
()
const
{
return
fpga_conv_args
;
}
void
SetFpgaArgs
(
const
fpga
::
Wrapper
ConvArgs
&
args
)
{
fpga_conv_args
=
args
;
}
void
SetFpgaArgs
(
const
fpga
::
Split
ConvArgs
&
args
)
{
fpga_conv_args
=
args
;
}
#endif
#endif
};
};
#endif
#endif
...
@@ -1857,11 +1857,11 @@ class FusionConvAddBNParam : public ConvParam<Dtype> {
...
@@ -1857,11 +1857,11 @@ class FusionConvAddBNParam : public ConvParam<Dtype> {
#ifdef PADDLE_MOBILE_FPGA
#ifdef PADDLE_MOBILE_FPGA
private:
private:
fpga
::
Wrapper
ConvArgs
fpga_conv_args
;
fpga
::
Split
ConvArgs
fpga_conv_args
;
public:
public:
const
fpga
::
Wrapper
ConvArgs
&
FpgaArgs
()
const
{
return
fpga_conv_args
;
}
const
fpga
::
Split
ConvArgs
&
FpgaArgs
()
const
{
return
fpga_conv_args
;
}
void
SetFpgaArgs
(
const
fpga
::
Wrapper
ConvArgs
&
args
)
{
fpga_conv_args
=
args
;
}
void
SetFpgaArgs
(
const
fpga
::
Split
ConvArgs
&
args
)
{
fpga_conv_args
=
args
;
}
#endif
#endif
};
};
#endif
#endif
...
@@ -1983,11 +1983,11 @@ class FusionConvBNReluParam : public ConvParam<Dtype> {
...
@@ -1983,11 +1983,11 @@ class FusionConvBNReluParam : public ConvParam<Dtype> {
#ifdef PADDLE_MOBILE_FPGA
#ifdef PADDLE_MOBILE_FPGA
private:
private:
fpga
::
Wrapper
ConvArgs
fpga_conv_args
;
fpga
::
Split
ConvArgs
fpga_conv_args
;
public:
public:
const
fpga
::
Wrapper
ConvArgs
&
FpgaArgs
()
const
{
return
fpga_conv_args
;
}
const
fpga
::
Split
ConvArgs
&
FpgaArgs
()
const
{
return
fpga_conv_args
;
}
void
SetFpgaArgs
(
const
fpga
::
Wrapper
ConvArgs
&
args
)
{
fpga_conv_args
=
args
;
}
void
SetFpgaArgs
(
const
fpga
::
Split
ConvArgs
&
args
)
{
fpga_conv_args
=
args
;
}
#endif
#endif
};
};
#endif
#endif
...
...
src/operators/sum_op.cpp
浏览文件 @
c4cce985
...
@@ -65,7 +65,6 @@ REGISTER_OPERATOR_CPU(sum, ops::SumOp);
...
@@ -65,7 +65,6 @@ REGISTER_OPERATOR_CPU(sum, ops::SumOp);
REGISTER_OPERATOR_MALI_GPU
(
sum
,
ops
::
ConcatOp
);
REGISTER_OPERATOR_MALI_GPU
(
sum
,
ops
::
ConcatOp
);
#endif
#endif
#ifdef PADDLE_MOBILE_FPGA
#ifdef PADDLE_MOBILE_FPGA
REGISTER_OPERATOR_FPGA
(
sum
,
ops
::
ConcatOp
);
#endif
#endif
#endif
#endif
test/CMakeLists.txt
浏览文件 @
c4cce985
...
@@ -61,38 +61,11 @@ endif ()
...
@@ -61,38 +61,11 @@ endif ()
list
(
FIND NET
"FPGAnets"
CON
)
list
(
FIND NET
"FPGAnets"
CON
)
if
(
CON GREATER -1
)
if
(
CON GREATER -1
)
ADD_EXECUTABLE
(
test-resnet net/test_resnet.cpp test_helper.h test_include.h executor_for_test.h
)
target_link_libraries
(
test-resnet paddle-mobile
)
ADD_EXECUTABLE
(
test-resnet50 fpga/test_resnet50.cpp test_helper.h test_include.h executor_for_test.h
)
ADD_EXECUTABLE
(
test-resnet50 fpga/test_resnet50.cpp test_helper.h test_include.h executor_for_test.h
)
target_link_libraries
(
test-resnet50 paddle-mobile
)
target_link_libraries
(
test-resnet50 paddle-mobile
)
ADD_EXECUTABLE
(
test-fpga-EW fpga/test_fpga_EW.cpp test_helper.h test_include.h executor_for_test.h
)
# ADD_EXECUTABLE(test-resnet net/test_resnet.cpp test_helper.h test_include.h executor_for_test.h)
target_link_libraries
(
test-fpga-EW paddle-mobile
)
# target_link_libraries(test-resnet paddle-mobile)
ADD_EXECUTABLE
(
test-fpga-conv fpga/test_fpga_conv.cpp test_helper.h test_include.h executor_for_test.h
)
target_link_libraries
(
test-fpga-conv paddle-mobile
)
ADD_EXECUTABLE
(
test-fpga-pooling fpga/test_fpga_pooling.cpp test_helper.h test_include.h executor_for_test.h
)
target_link_libraries
(
test-fpga-pooling paddle-mobile
)
ADD_EXECUTABLE
(
test-fpga-bypass fpga/test_fpga_bypass.cpp test_helper.h test_include.h executor_for_test.h
)
target_link_libraries
(
test-fpga-bypass paddle-mobile
)
ADD_EXECUTABLE
(
test-fpga-softmax fpga/test_fpga_softmax.cpp test_helper.h test_include.h executor_for_test.h
)
target_link_libraries
(
test-fpga-softmax paddle-mobile
)
ADD_EXECUTABLE
(
test-fpga-concat fpga/test_fpga_concat.cpp test_helper.h test_include.h executor_for_test.h
)
target_link_libraries
(
test-fpga-concat paddle-mobile
)
ADD_EXECUTABLE
(
test-tensor-quant fpga/test_tensor_quant.cpp test_helper.h test_include.h executor_for_test.h
)
target_link_libraries
(
test-tensor-quant paddle-mobile
)
ADD_EXECUTABLE
(
test-fpga-concat-op fpga/test_concat_op.cpp test_helper.h test_include.h
)
target_link_libraries
(
test-fpga-concat-op paddle-mobile
)
ADD_EXECUTABLE
(
test-format-data fpga/test_format_data.cpp test_helper.h test_include.h
)
target_link_libraries
(
test-format-data paddle-mobile
)
set
(
FOUND_MATCH ON
)
set
(
FOUND_MATCH ON
)
endif
()
endif
()
...
...
test/fpga/test_resnet50.cpp
浏览文件 @
c4cce985
...
@@ -30,7 +30,11 @@ int main() {
...
@@ -30,7 +30,11 @@ int main() {
input_tensor
.
data
<
float
>
()
+
input_tensor
.
numel
());
input_tensor
.
data
<
float
>
()
+
input_tensor
.
numel
());
paddle_mobile
.
FeedData
(
input_tensor
);
paddle_mobile
.
FeedData
(
input_tensor
);
paddle_mobile
.
Predict_To
(
-
1
);
for
(
int
i
=
0
;
i
<
1000
;
i
++
)
{
paddle_mobile
.
Predict_To
(
-
1
);
if
(
i
%
100
==
0
)
std
::
cout
<<
i
<<
std
::
endl
;
}
// paddle_mobile.Predict_From(73);
// paddle_mobile.Predict_From(73);
// paddle_mobile.Predict_From_To(72, 73);
// paddle_mobile.Predict_From_To(72, 73);
...
...
tools/op.cmake
浏览文件 @
c4cce985
...
@@ -118,12 +118,9 @@ if (CON GREATER -1)
...
@@ -118,12 +118,9 @@ if (CON GREATER -1)
set
(
POOL_OP ON
)
set
(
POOL_OP ON
)
set
(
CONCAT_OP ON
)
set
(
CONCAT_OP ON
)
set
(
SOFTMAX_OP ON
)
set
(
SOFTMAX_OP ON
)
set
(
DROPOUT_OP ON
)
set
(
FUSION_CONVBNRELU_OP ON
)
set
(
FUSION_CONVBNRELU_OP ON
)
set
(
FUSION_CONVBN_OP ON
)
set
(
FUSION_CONVBN_OP ON
)
set
(
FUSION_CONVADD_OP ON
)
set
(
FUSION_CONVADD_OP ON
)
set
(
MUL_OP ON
)
set
(
FOUND_MATCH ON
)
set
(
FOUND_MATCH ON
)
endif
()
endif
()
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录