Skip to content

GitLab

Commit 58990937 authored by Bartlomiej Kocot's avatar Bartlomiej Kocot
Browse files

Fix interface test

parent 68fda7c2
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
test/conv_tensor_rearrange/test_conv_tensor_rearrange_interface.cpp
View file @ 58990937
...@@ -53,7 +53,7 @@ class TestConvTensorRearrangeInterface : public ::testing::Test ...@@ -53,7 +53,7 @@ class TestConvTensorRearrangeInterface : public ::testing::Test
template <typename ConvTensorRearrangeOp> template <typename ConvTensorRearrangeOp>
bool Run() bool Run()
{ {
const auto G = conv_param.N_;
const auto N = conv_param.N_; const auto N = conv_param.N_;
const auto C = conv_param.C_; const auto C = conv_param.C_;
const auto FakeC = const auto FakeC =
...@@ -100,6 +100,7 @@ class TestConvTensorRearrangeInterface : public ::testing::Test ...@@ -100,6 +100,7 @@ class TestConvTensorRearrangeInterface : public ::testing::Test
auto img2col = DeviceImgToColInstance{}; auto img2col = DeviceImgToColInstance{};
auto argument = img2col.MakeArgument(nullptr, auto argument = img2col.MakeArgument(nullptr,
nullptr, nullptr,
G,
N, N,
IsCPacked ? C : FakeC, IsCPacked ? C : FakeC,
input_spatial_lengths, input_spatial_lengths,
...@@ -119,6 +120,7 @@ class TestConvTensorRearrangeInterface : public ::testing::Test ...@@ -119,6 +120,7 @@ class TestConvTensorRearrangeInterface : public ::testing::Test
auto col2img = DeviceColToimgInstance{}; auto col2img = DeviceColToimgInstance{};
auto argument = col2img.MakeArgument(nullptr, auto argument = col2img.MakeArgument(nullptr,
nullptr, nullptr,
G,
N, N,
IsCPacked ? C : FakeC, IsCPacked ? C : FakeC,
input_spatial_lengths, input_spatial_lengths,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment