Hi. It seems that recent commit breaks reduction in NNVM. (I’m now at current origin/master bde53033b). For example, my small argmax test program issues the below error. Please, consider fixing! What is the status of automatic tests currently?
TVMError Traceback (most recent call last)
<ipython-input-4-74ed2e3f3a21> in <module>()
----> 1 test_argmax()
~/src/mironov/argmax/test_argmax.py in test_argmax()
36 print('numpy', np.argmax(data+1, axis=0))
37 m.run(x=data)
---> 38 out = m.get_output(0, tvm.nd.empty(shape=oshape, dtype='int32'))
39 print('out',out)
40 return out
~/tvm/python/tvm/contrib/graph_runtime.py in get_output(self, index, out)
176 """
177 if out:
--> 178 self._get_output(index, out)
179 return out
180
~/tvm/python/tvm/_ffi/_ctypes/function.py in __call__(self, *args)
183 check_call(_LIB.TVMFuncCall(
184 self.handle, values, tcodes, ctypes.c_int(num_args),
--> 185 ctypes.byref(ret_val), ctypes.byref(ret_tcode)))
186 _ = temp_args
187 _ = args
~/tvm/python/tvm/_ffi/base.py in check_call(ret)
64 """
65 if ret != 0:
---> 66 raise TVMError(py_str(_LIB.TVMGetLastError()))
67
68
TVMError: [12:54:53] /workspace/tvm/src/runtime/graph/graph_runtime.cc:153: Check failed: data->shape[j] == data_out->shape[j] (1 vs. 4)