From 89798fd96ca87d42a4b6bc0623460e136cda0ea4 Mon Sep 17 00:00:00 2001
From: Wuwei Lin <vincentl13x@gmail.com>
Date: Tue, 9 Apr 2019 10:16:41 +0800
Subject: [PATCH] Return undef if original layout contains splitted axes

---
 src/relay/op/tensor/transform.cc | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/relay/op/tensor/transform.cc b/src/relay/op/tensor/transform.cc
index 593aa3bc86ef6..b227577a012f3 100644
--- 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]);