Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- TVMError Traceback (most recent call last)
- <ipython-input-6-57e582369e16> in <module>
- 1 with tvm.transform.PassContext(opt_level=3, config={'tir.disable_vectorize': True}):
- ----> 2 result = tvm.relay.build(relay_model, target=TARGET, params=params)
- 3 graph_json, compiled_model, simplified_params = result.get_json(), result.get_lib(), result.get_params()
- /home/user/TVM/tvm/python/tvm/relay/build_module.py in build(ir_mod, target, target_host, executor, runtime, workspace_memory_pools, params, mod_name)
- 475 runtime=runtime,
- 476 workspace_memory_pools=workspace_memory_pools,
- --> 477 mod_name=mod_name,
- 478 )
- 479 func_metadata = bld_mod.get_function_metadata()
- /home/user/TVM/tvm/python/tvm/relay/build_module.py in build(self, mod, target, target_host, executor, runtime, workspace_memory_pools, params, mod_name)
- 192 mod_name = mangle_module_name(mod_name)
- 193
- --> 194 self._build(mod, raw_targets, executor, runtime, workspace_memory_pools, mod_name)
- 195 autotvm.GLOBAL_SCOPE.silent = old_autotvm_silent
- 196
- /home/user/TVM/tvm/python/tvm/_ffi/_ctypes/packed_func.py in __call__(self, *args)
- 235 != 0
- 236 ):
- --> 237 raise get_last_ffi_error()
- 238 _ = temp_args
- 239 _ = args
- TVMError: Traceback (most recent call last):
- 34: TVMFuncCall
- 33: tvm::relay::backend::RelayBuildModule::GetFunction(std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> > const&, tvm::runtime::ObjectPtr<tvm::runtime::Object> const&)::{lambda(tvm::runtime::TVMArgs, tvm::runtime::TVMRetValue*)#3}::operator()(tvm::runtime::TVMArgs, tvm::runtime::TVMRetValue*) const
- 32: tvm::relay::backend::RelayBuildModule::BuildRelay(tvm::IRModule, tvm::runtime::String const&)
- 31: tvm::runtime::PackedFuncObj::Extractor<tvm::runtime::PackedFuncSubObj<tvm::relay::backend::GraphExecutorCodegenModule::GetFunction(std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> > const&, tvm::runtime::ObjectPtr<tvm::runtime::Object> const&)::{lambda(tvm::runtime::TVMArgs, tvm::runtime::TVMRetValue*)#2}> >::Call(tvm::runtime::PackedFuncObj const*, tvm::runtime::TVMArgs, tvm::runtime::TVMRetValue*)
- 30: tvm::relay::backend::GraphExecutorCodegen::Codegen(tvm::IRModule, tvm::relay::Function, tvm::runtime::String)
- 29: tvm::transform::Pass::operator()(tvm::IRModule) const
- 28: tvm::transform::Pass::operator()(tvm::IRModule, tvm::transform::PassContext const&) const
- 27: tvm::transform::SequentialNode::operator()(tvm::IRModule, tvm::transform::PassContext const&) const
- 26: tvm::transform::Pass::operator()(tvm::IRModule, tvm::transform::PassContext const&) const
- 25: tvm::transform::ModulePassNode::operator()(tvm::IRModule, tvm::transform::PassContext const&) const
- 24: _ZN3tvm7runtime13PackedFuncObj9ExtractorINS0_16PackedFuncSubObjIZNS0_15TypedPackedFuncIFNS_8IRModuleES5_NS_9transform11PassContextEEE17AssignTypedLambdaIZNS_5relay3tec11LowerTEPassERKNS0_6StringESt8functionIFvNS_8BaseFuncEEENS_13VirtualDeviceEEUlS5_S7_E_EEvT_EUlRKNS0_7TVMArgsEPNS0_11TVMRetValueEE_EEE4CallEPKS1_SN_SR_
- 23: tvm::relay::tec::LowerTE(tvm::IRModule const&, tvm::runtime::String const&, std::function<void (tvm::BaseFunc)>, tvm::VirtualDevice)
- 22: tvm::transform::Pass::operator()(tvm::IRModule) const
- 21: tvm::transform::Pass::operator()(tvm::IRModule, tvm::transform::PassContext const&) const
- 20: tvm::relay::transform::FunctionPassNode::operator()(tvm::IRModule, tvm::transform::PassContext const&) const
- 19: _ZN3tvm7runtime13PackedFuncObj9ExtractorINS0_16PackedFuncSubObjIZNS0_15TypedPackedFuncIFNS_5relay8FunctionES6_NS_8IRModuleENS_9transform11PassContextEEE17AssignTypedLambdaIZNS5_3tec15LowerTensorExprERKNS0_6StringENSD_10TECompilerESt8functionIFvNS_8BaseFuncEEENS_13VirtualDeviceEEUlS6_S7_S9_E_EEvT_EUlRKNS0_7TVMArgsEPNS0_11TVMRetValueEE_EEE4CallEPKS1_SP_ST_
- 18: tvm::relay::ExprMutator::VisitExpr(tvm::RelayExpr const&)
- 17: tvm::relay::ExprFunctor<tvm::RelayExpr (tvm::RelayExpr const&)>::VisitExpr(tvm::RelayExpr const&)
- 16: _ZZN3tvm5relay11ExprFunctorIFNS_9RelayExprERKS2_EE10InitVTableEvENUlR
- 15: tvm::relay::transform::DeviceAwareExprMutator::VisitExpr_(tvm::relay::FunctionNode const*)
- 14: tvm::relay::tec::LowerTensorExprMutator::DeviceAwareVisitExpr_(tvm::relay::FunctionNode const*)
- 13: _ZN3tvm5relay9transform22DeviceAwareExprMutator21DeviceAwar
- 12: tvm::relay::ExprMutator::VisitExpr_(tvm::relay::FunctionNode const*)
- 11: tvm::relay::ExprMutator::VisitExpr(tvm::RelayExpr const&)
- 10: tvm::relay::ExprFunctor<tvm::RelayExpr (tvm::RelayExpr const&)>::VisitExpr(tvm::RelayExpr const&)
- 9: _ZZN3tvm5relay11ExprFunctorIFNS_9RelayExprERKS2_EE10InitVTableEvENUlR
- 8: tvm::relay::transform::DeviceAwareExprMutator::VisitExpr_(tvm::relay::CallNode const*)
- 7: tvm::relay::tec::LowerTensorExprMutator::DeviceAwareVisitExpr_(tvm::relay::CallNode const*)
- 6: tvm::relay::tec::LowerTensorExprMutator::MakeLoweredCall(tvm::relay::Function, tvm::runtime::Array<tvm::RelayExpr, void>, tvm::Span, tvm::Target)
- 5: tvm::relay::tec::TECompilerImpl::Lower(tvm::relay::tec::CCacheKey const&, tvm::runtime::String)
- 4: tvm::relay::tec::TECompilerImpl::LowerInternal(tvm::relay::tec::CCacheKey const&, std::function<tvm::runtime::String (tvm::runtime::String)>)
- 3: tvm::relay::tec::PrimFuncFor(tvm::relay::Function const&, tvm::Target const&, std::function<std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> > (std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> >)>)
- 2: tvm::relay::tec::ScheduleBuilder::Create(tvm::relay::Function const&, std::function<std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> > (std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> >)>)
- 1: tvm::relay::OpImplementation::Schedule(tvm::Attrs const&, tvm::runtime::Array<tvm::te::Tensor, void> const&, tvm::Target const&)
- 0: tvm::runtime::PackedFuncObj::Extractor<tvm::runtime::PackedFuncSubObj<TVMFuncCreateFromCFunc::{lambda(tvm::runtime::TVMArgs, tvm::runtime::TVMRetValue*)#2}> >::Call(tvm::runtime::PackedFuncObj const*, tvm::runtime::TVMArgs, tvm::runtime::TVMRetValue*)
- File "/home/user/TVM/tvm/python/tvm/_ffi/_ctypes/packed_func.py", line 81, in cfun
- rv = local_pyfunc(*pyargs)
- File "/home/user/TVM/tvm/python/tvm/relay/op/strategy/generic.py", line 51, in wrapper
- return topi_schedule(outs)
- File "/home/user/TVM/tvm/python/tvm/topi/arm_cpu/dense.py", line 25, in schedule_dense_dsp
- return dense_dsp_schedule(outs)
- File "/home/user/TVM/tvm/python/tvm/topi/arm_cpu/mprofile/dsp/dense.py", line 51, in dense_dsp_schedule
- traverse_inline(sched, outs[-1].op, _callback)
- File "/home/user/TVM/tvm/python/tvm/topi/utils.py", line 81, in traverse_inline
- _traverse(final_op)
- File "/home/user/TVM/tvm/python/tvm/topi/utils.py", line 78, in _traverse
- _traverse(tensor.op)
- File "/home/user/TVM/tvm/python/tvm/topi/utils.py", line 78, in _traverse
- _traverse(tensor.op)
- File "/home/user/TVM/tvm/python/tvm/topi/utils.py", line 78, in _traverse
- _traverse(tensor.op)
- [Previous line repeated 1 more time]
- File "/home/user/TVM/tvm/python/tvm/topi/utils.py", line 79, in _traverse
- callback(op)
- File "/home/user/TVM/tvm/python/tvm/topi/arm_cpu/mprofile/dsp/dense.py", line 47, in _callback
- gemm, uniq_id = intrin_gemm_MxKxN(M, K, N, data_vec.dtype, output.dtype)
- File "/home/user/TVM/tvm/python/tvm/topi/arm_cpu/mprofile/dsp/micro_kernel/gemm.py", line 53, in intrin_gemm_MxKxN
- assert out_dtype == "int32"
- TVMError: AssertionError
Advertisement
Add Comment
Please, Sign In to add comment