ctest 4

Tencent · Jan 13, 2025 · 0a3aa28 · 0a3aa28
1 parent 5be32e1
commit 0a3aa28
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/layer/flip.cpp b/src/layer/flip.cpp
@@ -116,6 +116,8 @@ int Flip::forward(const Mat& bottom_blob, Mat& top_blob, const Option& opt) cons
     else if (dims == 3)
     {
         top_blob.create(w, h, channels, elemsize, opt.blob_allocator);
+        if (top_blob.empty())
+            return -100;
         if (axis.w == 1)
         {
             // w、h、c