diff --git a/ynnpack/subgraph/runtime.cc b/ynnpack/subgraph/runtime.cc index 11489eecb87..f39c0a4894e 100644 --- a/ynnpack/subgraph/runtime.cc +++ b/ynnpack/subgraph/runtime.cc @@ -129,6 +129,7 @@ std::unique_ptr ynn_runtime::make_schedule( for (int index_d = 0; index_d < rank; ++index_d) { int d = get_loop_dim(index_d); if (extents[d].defined() && splits[d].defined()) { + if (slinky::prove_true(extents[d] == 1)) continue; loop_splits.push_back({dims[d], splits[d], workers[d], extents[d]}); } }