Return undef if original layout contains splitted axes

apache · Apr 9, 2019 · 89798fd · 89798fd
1 parent d1056fc
commit 89798fd
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/relay/op/tensor/transform.cc b/src/relay/op/tensor/transform.cc
@@ -1726,7 +1726,10 @@ Array<Array<Layout> > StridedSliceInferCorrectLayout(
 
     for (size_t i = 0; i < params->begin.size(); i++) {
       const LayoutAxis& axis = layout[i];
-      CHECK(axis.IsPrimal());
+      if (!axis.IsPrimal()) {
+        // original layout that contains splitted axes is not supported
+        return {{Layout::Undef()}, {Layout::Undef()}};
+      }
       auto factor = new_layout.FactorOf(axis);
       if (factor == -1) {
         new_begin.push_back(params->begin[i]);