提交 78326ee6 编写于 作者: W wuchenghui

fix concat benchmark

上级 77b3f45d
...@@ -91,13 +91,13 @@ static void OpenclConcatHelper(int iters, ...@@ -91,13 +91,13 @@ static void OpenclConcatHelper(int iters,
} }
#define BM_CONCAT_OPENCL_MACRO(N, C, H, W, TYPE) \ #define BM_CONCAT_OPENCL_MACRO(N, H, W, C, TYPE) \
static void BM_CONCAT_OPENCL_##N##_##C##_##H##_##W##_##TYPE( \ static void BM_CONCAT_OPENCL_##N##_##H##_##W##_##C##_##TYPE( \
int iters) { \ int iters) { \
std::vector<index_t> shape = {N, H, W, C}; \ std::vector<index_t> shape = {N, H, W, C}; \
OpenclConcatHelper<TYPE>(iters, shape, shape, 3); \ OpenclConcatHelper<TYPE>(iters, shape, shape, 3); \
} \ } \
BENCHMARK(BM_CONCAT_OPENCL_##N##_##C##_##H##_##W##_##TYPE) BENCHMARK(BM_CONCAT_OPENCL_##N##_##H##_##W##_##C##_##TYPE)
BM_CONCAT_OPENCL_MACRO(3, 32, 32, 32, float); BM_CONCAT_OPENCL_MACRO(3, 32, 32, 32, float);
BM_CONCAT_OPENCL_MACRO(3, 32, 32, 64, float); BM_CONCAT_OPENCL_MACRO(3, 32, 32, 64, float);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册