don't allow empty output from arange

apache · Feb 20, 2019 · 4ed56e9 · 4ed56e9
1 parent 017a2c3
commit 4ed56e9
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/src/relay/op/tensor/transform.cc b/src/relay/op/tensor/transform.cc
@@ -892,7 +892,7 @@ bool ArangeRel(const Array<Type>& types,
   IndexExpr num_elem = tvm::cast(tvm::Int(32), tvm::ceil(
       tvm::cast(tvm::Float(32), param->stop - param->start) / param->step));
   if (const tvm::ir::IntImm* val = num_elem.as<tvm::ir::IntImm>()) {
-    CHECK_GE(val->value, 0) << "Invalid arange inputs";
+    CHECK_GT(val->value, 0) << "Invalid arange inputs";
   }
   reporter->Assign(types[0], TensorTypeNode::make({num_elem}, param->dtype));
   return true;

diff --git a/tests/python/relay/test_op_level3.py b/tests/python/relay/test_op_level3.py
@@ -489,7 +489,6 @@ def verify_arange(start, stop, step):
     verify_arange(1, 20, 3)
     verify_arange(20, 1, -1)
     verify_arange(20, 1, -1.5)
-    verify_arange(5, 5, 1)
 
 
 if __name__ == "__main__":