update check non negative

Oneflow-Inc · May 29, 2024 · b0881a8 · b0881a8
1 parent 0a78cdc
commit b0881a8
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/oneflow/core/functional/impl/common.cpp b/oneflow/core/functional/impl/common.cpp
@@ -206,9 +206,8 @@ Maybe<void> CheckInplaceShapeCanExpandTo(const Shape& shape, const Shape& expand
 
 Maybe<void> CheckSizeNonNegative(const Shape& shape) {
   for (const auto& s : shape) {
-    CHECK_OR_THROW(s >= 0)
-        << "Trying to create tensor with negative dimension " << s << ": "
-        << shape;
+    CHECK_OR_RETURN(s >= 0) << "Trying to create tensor with negative dimension " << s << ": "
+                            << shape;
   }
   return Maybe<void>::Ok();
 }