[nnvm] fix nnvm compiler build module error (apache#3378)

neo-ai · Jun 27, 2019 · 1069eee · 1069eee
1 parent 0842d4e
commit 1069eee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nnvm/python/nnvm/compiler/build_module.py b/nnvm/python/nnvm/compiler/build_module.py
@@ -171,7 +171,7 @@ def _update_shape_dtype(shape, dtype, params):
     shape.update({k : v.shape for k, v in params.items()})
     if isinstance(dtype, str):
         for k, v in params.items():
-            if v.dtype != dtype:
+            if v.dtype != dtype and v.shape:
                 raise ValueError(
                     "%s: dtype not expected %s vs %s" % (k, dtype, v.dtype))
     else: