nrdmttt

uTVM error

Jun 24th, 2022
46
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 7.65 KB | None | 0 0
  1. TVMError Traceback (most recent call last)
  2. <ipython-input-6-57e582369e16> in <module>
  3. 1 with tvm.transform.PassContext(opt_level=3, config={'tir.disable_vectorize': True}):
  4. ----> 2 result = tvm.relay.build(relay_model, target=TARGET, params=params)
  5. 3 graph_json, compiled_model, simplified_params = result.get_json(), result.get_lib(), result.get_params()
  6.  
  7. /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)
  8. 475 runtime=runtime,
  9. 476 workspace_memory_pools=workspace_memory_pools,
  10. --> 477 mod_name=mod_name,
  11. 478 )
  12. 479 func_metadata = bld_mod.get_function_metadata()
  13.  
  14. /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)
  15. 192 mod_name = mangle_module_name(mod_name)
  16. 193
  17. --> 194 self._build(mod, raw_targets, executor, runtime, workspace_memory_pools, mod_name)
  18. 195 autotvm.GLOBAL_SCOPE.silent = old_autotvm_silent
  19. 196
  20.  
  21. /home/user/TVM/tvm/python/tvm/_ffi/_ctypes/packed_func.py in __call__(self, *args)
  22. 235 != 0
  23. 236 ):
  24. --> 237 raise get_last_ffi_error()
  25. 238 _ = temp_args
  26. 239 _ = args
  27.  
  28. TVMError: Traceback (most recent call last):
  29. 34: TVMFuncCall
  30. 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
  31. 32: tvm::relay::backend::RelayBuildModule::BuildRelay(tvm::IRModule, tvm::runtime::String const&)
  32. 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*)
  33. 30: tvm::relay::backend::GraphExecutorCodegen::Codegen(tvm::IRModule, tvm::relay::Function, tvm::runtime::String)
  34. 29: tvm::transform::Pass::operator()(tvm::IRModule) const
  35. 28: tvm::transform::Pass::operator()(tvm::IRModule, tvm::transform::PassContext const&) const
  36. 27: tvm::transform::SequentialNode::operator()(tvm::IRModule, tvm::transform::PassContext const&) const
  37. 26: tvm::transform::Pass::operator()(tvm::IRModule, tvm::transform::PassContext const&) const
  38. 25: tvm::transform::ModulePassNode::operator()(tvm::IRModule, tvm::transform::PassContext const&) const
  39. 24: _ZN3tvm7runtime13PackedFuncObj9ExtractorINS0_16PackedFuncSubObjIZNS0_15TypedPackedFuncIFNS_8IRModuleES5_NS_9transform11PassContextEEE17AssignTypedLambdaIZNS_5relay3tec11LowerTEPassERKNS0_6StringESt8functionIFvNS_8BaseFuncEEENS_13VirtualDeviceEEUlS5_S7_E_EEvT_EUlRKNS0_7TVMArgsEPNS0_11TVMRetValueEE_EEE4CallEPKS1_SN_SR_
  40. 23: tvm::relay::tec::LowerTE(tvm::IRModule const&, tvm::runtime::String const&, std::function<void (tvm::BaseFunc)>, tvm::VirtualDevice)
  41. 22: tvm::transform::Pass::operator()(tvm::IRModule) const
  42. 21: tvm::transform::Pass::operator()(tvm::IRModule, tvm::transform::PassContext const&) const
  43. 20: tvm::relay::transform::FunctionPassNode::operator()(tvm::IRModule, tvm::transform::PassContext const&) const
  44. 19: _ZN3tvm7runtime13PackedFuncObj9ExtractorINS0_16PackedFuncSubObjIZNS0_15TypedPackedFuncIFNS_5relay8FunctionES6_NS_8IRModuleENS_9transform11PassContextEEE17AssignTypedLambdaIZNS5_3tec15LowerTensorExprERKNS0_6StringENSD_10TECompilerESt8functionIFvNS_8BaseFuncEEENS_13VirtualDeviceEEUlS6_S7_S9_E_EEvT_EUlRKNS0_7TVMArgsEPNS0_11TVMRetValueEE_EEE4CallEPKS1_SP_ST_
  45. 18: tvm::relay::ExprMutator::VisitExpr(tvm::RelayExpr const&)
  46. 17: tvm::relay::ExprFunctor<tvm::RelayExpr (tvm::RelayExpr const&)>::VisitExpr(tvm::RelayExpr const&)
  47. 16: _ZZN3tvm5relay11ExprFunctorIFNS_9RelayExprERKS2_EE10InitVTableEvENUlR
  48. 15: tvm::relay::transform::DeviceAwareExprMutator::VisitExpr_(tvm::relay::FunctionNode const*)
  49. 14: tvm::relay::tec::LowerTensorExprMutator::DeviceAwareVisitExpr_(tvm::relay::FunctionNode const*)
  50. 13: _ZN3tvm5relay9transform22DeviceAwareExprMutator21DeviceAwar
  51. 12: tvm::relay::ExprMutator::VisitExpr_(tvm::relay::FunctionNode const*)
  52. 11: tvm::relay::ExprMutator::VisitExpr(tvm::RelayExpr const&)
  53. 10: tvm::relay::ExprFunctor<tvm::RelayExpr (tvm::RelayExpr const&)>::VisitExpr(tvm::RelayExpr const&)
  54. 9: _ZZN3tvm5relay11ExprFunctorIFNS_9RelayExprERKS2_EE10InitVTableEvENUlR
  55. 8: tvm::relay::transform::DeviceAwareExprMutator::VisitExpr_(tvm::relay::CallNode const*)
  56. 7: tvm::relay::tec::LowerTensorExprMutator::DeviceAwareVisitExpr_(tvm::relay::CallNode const*)
  57. 6: tvm::relay::tec::LowerTensorExprMutator::MakeLoweredCall(tvm::relay::Function, tvm::runtime::Array<tvm::RelayExpr, void>, tvm::Span, tvm::Target)
  58. 5: tvm::relay::tec::TECompilerImpl::Lower(tvm::relay::tec::CCacheKey const&, tvm::runtime::String)
  59. 4: tvm::relay::tec::TECompilerImpl::LowerInternal(tvm::relay::tec::CCacheKey const&, std::function<tvm::runtime::String (tvm::runtime::String)>)
  60. 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> >)>)
  61. 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> >)>)
  62. 1: tvm::relay::OpImplementation::Schedule(tvm::Attrs const&, tvm::runtime::Array<tvm::te::Tensor, void> const&, tvm::Target const&)
  63. 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*)
  64. File "/home/user/TVM/tvm/python/tvm/_ffi/_ctypes/packed_func.py", line 81, in cfun
  65. rv = local_pyfunc(*pyargs)
  66. File "/home/user/TVM/tvm/python/tvm/relay/op/strategy/generic.py", line 51, in wrapper
  67. return topi_schedule(outs)
  68. File "/home/user/TVM/tvm/python/tvm/topi/arm_cpu/dense.py", line 25, in schedule_dense_dsp
  69. return dense_dsp_schedule(outs)
  70. File "/home/user/TVM/tvm/python/tvm/topi/arm_cpu/mprofile/dsp/dense.py", line 51, in dense_dsp_schedule
  71. traverse_inline(sched, outs[-1].op, _callback)
  72. File "/home/user/TVM/tvm/python/tvm/topi/utils.py", line 81, in traverse_inline
  73. _traverse(final_op)
  74. File "/home/user/TVM/tvm/python/tvm/topi/utils.py", line 78, in _traverse
  75. _traverse(tensor.op)
  76. File "/home/user/TVM/tvm/python/tvm/topi/utils.py", line 78, in _traverse
  77. _traverse(tensor.op)
  78. File "/home/user/TVM/tvm/python/tvm/topi/utils.py", line 78, in _traverse
  79. _traverse(tensor.op)
  80. [Previous line repeated 1 more time]
  81. File "/home/user/TVM/tvm/python/tvm/topi/utils.py", line 79, in _traverse
  82. callback(op)
  83. File "/home/user/TVM/tvm/python/tvm/topi/arm_cpu/mprofile/dsp/dense.py", line 47, in _callback
  84. gemm, uniq_id = intrin_gemm_MxKxN(M, K, N, data_vec.dtype, output.dtype)
  85. File "/home/user/TVM/tvm/python/tvm/topi/arm_cpu/mprofile/dsp/micro_kernel/gemm.py", line 53, in intrin_gemm_MxKxN
  86. assert out_dtype == "int32"
  87. TVMError: AssertionError
Advertisement
Add Comment
Please, Sign In to add comment