From: Mengqing Cao <redacted>
Date: Fri, 2 Aug 2024 08:50:53 +0000 (+0800)
Subject: cann: Fix ggml_cann_im2col for 1D im2col (llama/8819)
X-Git-Tag: upstream/0.0.1642~471
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=0a19c02501ca009b2c832e5f0c71cd3e3f86ccd0;p=pkg%2Fggml%2Fsources%2Fggml

cann: Fix ggml_cann_im2col for 1D im2col (llama/8819)

* fix ggml_cann_im2col for 1D im2col

* fix build warning
---

diff --git a/tests/test-backend-ops.cpp b/tests/test-backend-ops.cpp
index 5de70d55..f5065f14 100644
--- a/tests/test-backend-ops.cpp
+++ b/tests/test-backend-ops.cpp
@@ -2139,6 +2139,9 @@ static bool test_backend(ggml_backend_t backend, test_mode mode, const char * op
 
     test_cases.emplace_back(new test_im2col(GGML_TYPE_F32, GGML_TYPE_F16, GGML_TYPE_F32));
     test_cases.emplace_back(new test_im2col(GGML_TYPE_F32, GGML_TYPE_F16, GGML_TYPE_F16));
+    // test cases for 1D im2col
+    test_cases.emplace_back(new test_im2col(GGML_TYPE_F32, GGML_TYPE_F16, GGML_TYPE_F16, {3000, 128, 1, 1}, {3, 128, 1280, 1}, 1, 0, 1, 0, 1, 0, false));
+    test_cases.emplace_back(new test_im2col(GGML_TYPE_F32, GGML_TYPE_F16, GGML_TYPE_F32, {3000, 128, 1, 1}, {3, 128, 1280, 1}, 1, 0, 1, 0, 1, 0, false));
 
     test_cases.emplace_back(new test_conv_transpose_1d());
     test_cases.emplace_back(new test_conv_transpose_1d({3,2,1,1}, {2,3,2,1}, 3, 0, 1));