Commit 1e40e0c0 by Alexander Pivovarov Committed by Yizhi Liu

Allow long type values in shape list (#1806)

* Allow long type values in shape list

* Update build_module.py
parent e62d909a
......@@ -251,8 +251,8 @@ def build(graph, target=None, shape=None, dtype="float32",
if not isinstance(shape, dict):
raise TypeError("require shape to be dict")
for value in shape.values():
if not all(isinstance(x, int) for x in value):
raise TypeError("shape value must be int iterator")
if not all(isinstance(x, tvm._ffi.base.integer_types) for x in value):
raise TypeError("shape value must be Integer types iterator")
cfg = BuildConfig.current
graph = graph if isinstance(graph, _graph.Graph) else _graph.create(graph)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment