From 1529bce5259faa108884abc74031a2f49f4ee1ae Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Mon, 22 Aug 2022 19:23:39 +0800
Subject: [PATCH] perf(opencl): add opencl weight transpose kernel

GitOrigin-RevId: 1102ecf4bb602f37d1335007bf40c4b145b0a4a4
---
 dnn/test/common/topk.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dnn/test/common/topk.h b/dnn/test/common/topk.h
index 0bd5baebf..435759cde 100644
--- a/dnn/test/common/topk.h
+++ b/dnn/test/common/topk.h
@@ -16,7 +16,7 @@ private:
 public:
     OprProxy() = default;
     OprProxy(int k) : m_k{k} {}
-    void init(TopK*, const TensorLayoutArray&) {}
+    void init(TopK*, const TensorNDArray&) {}
 
     void deduce_layout(TopK* opr, TensorLayoutArray& layouts) {
         if (layouts.size() == 3) {
-- 
GitLab