From de084f92ba7780f63cf86174b05f56f34fd46181 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Tue, 18 Jul 2023 10:43:09 +0800
Subject: [PATCH] fix(src/atlas): remove the limitation that the input of the
 om model must be 4-dimensional

GitOrigin-RevId: ab3fce058e8ce1fa47e05142170b06a33bdcbf95
---
 src/opr/impl/atlas_runtime_op.cpp | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/src/opr/impl/atlas_runtime_op.cpp b/src/opr/impl/atlas_runtime_op.cpp
index 89ca0ba77..0d15ad6e3 100644
--- a/src/opr/impl/atlas_runtime_op.cpp
+++ b/src/opr/impl/atlas_runtime_op.cpp
@@ -40,9 +40,6 @@ TensorShape acl_shape_to_mgb_shape_for_input(
     MGB_MARK_USED_VAR(aipp_input_fmt);
     TensorShape ret;
     ret.ndim = acl_shape.dimCount;
-    mgb_assert(
-            ret.ndim == 4, "Unexpected ndim form aclmdlIODims expected 4, but got %zu",
-            ret.ndim);
     for (size_t i = 0; i < ret.ndim; ++i) {
         ret[i] = acl_shape.dims[i];
     }
-- 
GitLab