Fix interface test

58990937 · Bartlomiej Kocot · 68fda7c2 · 58990937
Commit 58990937 authored Oct 25, 2023 by Bartlomiej Kocot
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

test/conv_tensor_rearrange/test_conv_tensor_rearrange_interface.cpp ...tensor_rearrange/test_conv_tensor_rearrange_interface.cpp +3 -1

No files found.
--- a/test/conv_tensor_rearrange/test_conv_tensor_rearrange_interface.cpp
+++ b/test/conv_tensor_rearrange/test_conv_tensor_rearrange_interface.cpp
@@ -53,7 +53,7 @@ class TestConvTensorRearrangeInterface : public ::testing::Test
    template <typename ConvTensorRearrangeOp>
    bool Run()
    {
-
+        const auto G = conv_param.N_;
        const auto N = conv_param.N_;
        const auto C = conv_param.C_;
        const auto FakeC =
@@ -100,6 +100,7 @@ class TestConvTensorRearrangeInterface : public ::testing::Test
            auto img2col  = DeviceImgToColInstance{};
            auto argument = img2col.MakeArgument(nullptr,
                                                 nullptr,
+                                                 G,
                                                 N,
                                                 IsCPacked ? C : FakeC,
                                                 input_spatial_lengths,
@@ -119,6 +120,7 @@ class TestConvTensorRearrangeInterface : public ::testing::Test
            auto col2img  = DeviceColToimgInstance{};
            auto argument = col2img.MakeArgument(nullptr,
                                                 nullptr,
+                                                 G,
                                                 N,
                                                 IsCPacked ? C : FakeC,
                                                 input_spatial_lengths,