Skip to content

Mace
Mace

慢慢CG / Mace
与 Fork 源项目一致

Fork自 Xiaomi / Mace

通知 1
Star 0
Fork 0
Mace
Mace
提交 3af4e80b 编写于 作者: L Liangliang He
浏览文件
操作

Support half

上级 3d06f30c
隐藏空白更改
内联 并排
Showing with 8 addition and 2 deletion
mace/ops/addn_benchmark.cc
浏览文件 @ 3af4e80b
...@@ -29,13 +29,17 @@ static void AddNBenchmark(int iters, int inputs, int n, int h, int w, int c) { ...@@ -29,13 +29,17 @@ static void AddNBenchmark(int iters, int inputs, int n, int h, int w, int c) {
for (int i = 0; i < inputs; ++i) { for (int i = 0; i < inputs; ++i) {
op_def_builder.Input(internal::MakeString("InputImage", i).c_str()); op_def_builder.Input(internal::MakeString("InputImage", i).c_str());
} }
op_def_builder.Output("OutputImage").Finalize(net.NewOperatorDef()); op_def_builder.Output("OutputImage")
.AddIntArg("T", static_cast<int>(DataTypeToEnum<T>::value))
.Finalize(net.NewOperatorDef());
} else { } else {
OpDefBuilder op_def_builder("AddN", "AddNBM"); OpDefBuilder op_def_builder("AddN", "AddNBM");
for (int i = 0; i < inputs; ++i) { for (int i = 0; i < inputs; ++i) {
op_def_builder.Input(internal::MakeString("Input", i).c_str()); op_def_builder.Input(internal::MakeString("Input", i).c_str());
} }
op_def_builder.Output("Output").Finalize(net.NewOperatorDef()); op_def_builder.Output("Output")
.AddIntArg("T", static_cast<int>(DataTypeToEnum<T>::value))
.Finalize(net.NewOperatorDef());
} }
// Warm-up // Warm-up
...@@ -66,6 +70,8 @@ static void AddNBenchmark(int iters, int inputs, int n, int h, int w, int c) { ...@@ -66,6 +70,8 @@ static void AddNBenchmark(int iters, int inputs, int n, int h, int w, int c) {
BM_ADDN_MACRO(INPUTS, N, H, W, C, TYPE, OPENCL); BM_ADDN_MACRO(INPUTS, N, H, W, C, TYPE, OPENCL);
BM_ADDN(2, 1, 240, 240, 256, float); BM_ADDN(2, 1, 240, 240, 256, float);
// BM_ADDN(2, 1, 240, 240, 256, half);
BM_ADDN(4, 1, 240, 240, 256, float); BM_ADDN(4, 1, 240, 240, 256, float);
// BM_ADDN(4, 1, 240, 240, 256, half);
} // namespace mace } // namespace mace
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或