fix

Oneflow-Inc · Aug 25, 2023 · 99b3d60 · 99b3d60
1 parent 0aab78e
commit 99b3d60
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/oneflow/core/functional/impl/nn_functor.cpp b/oneflow/core/functional/impl/nn_functor.cpp
@@ -5223,6 +5223,7 @@ class GroupedMatmulFunctor {
  Maybe<TensorTuple> operator()(const TensorTuple& xs, const TensorTuple& weights) const {
  const int64_t input_size = xs.size();
  const int64_t weight_size = weights.size();
+ CHECK_LT_OR_RETURN(input_size, kMaxInputCount);
  CHECK_GE_OR_RETURN(input_size, 1)
  << Error::RuntimeError() << "The number of xs should be greater equal than 1.";
  CHECK_EQ_OR_RETURN(weight_size, input_size)