Skip to content

Commit

Permalink
Fix some Mat constructor calling
Browse files Browse the repository at this point in the history
  • Loading branch information
daquexian committed Aug 21, 2019
1 parent 56f39d3 commit 8877654
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions benchmark/benchmark.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -14,17 +14,17 @@
#include <dabnn/net.h>

static void BM_pack_mat_64_small(benchmark::State &state) {
const bnn::Mat a(1, 32, 32, 128, bnn::DataType::Float, 0);
bnn::Mat b(1, 32, 32, 128, bnn::DataType::Bit, 0);
const bnn::Mat a(1, 32, 32, 128, bnn::DataType::Float, false);
bnn::Mat b(1, 32, 32, 128, bnn::DataType::Bit, false);
for (auto _ : state) {
pack_mat_64(a, b);
}
}

#ifdef __aarch64__
static void BM_pack_mat_128_small(benchmark::State &state) {
const bnn::Mat a(1, 32, 32, 128, bnn::DataType::Float, 0);
bnn::Mat b(1, 32, 32, 128, bnn::DataType::Bit, 0);
const bnn::Mat a(1, 32, 32, 128, bnn::DataType::Float, false);
bnn::Mat b(1, 32, 32, 128, bnn::DataType::Bit, false);
for (auto _ : state) {
pack_mat_128(a, b);
}
Expand Down

0 comments on commit 8877654

Please sign in to comment.