I have used the below mentioned code and library to generate the .onnx file and when i try compiling I get the error : RuntimeError: Invalid type of axis: <class ‘tvm.tir.expr.Any’>. Could you please help me out ?
The complete error :
/home/kausthub/Desktop/major-project/env/lib/python3.8/site-packages/tvm/relay/frontend/onnx.py:5538: UserWarning: Input input_1 has unknown dimension shapes: ['unk__67', 'unk__68', 'unk__69', 1]. Specifying static values may improve performance
warnings.warn(warning_msg)
/home/kausthub/Desktop/major-project/env/lib/python3.8/site-packages/tvm/driver/build_module.py:267: UserWarning: target_host parameter is going to be deprecated. Please pass in tvm.target.Target(target, host=target_host) instead.
warnings.warn(
Traceback (most recent call last):
File "/home/kausthub/Desktop/major-project/env/bin/tvmc", line 8, in <module>
sys.exit(main())
File "/home/kausthub/Desktop/major-project/env/lib/python3.8/site-packages/tvm/driver/tvmc/main.py", line 115, in main
sys.exit(_main(sys.argv[1:]))
File "/home/kausthub/Desktop/major-project/env/lib/python3.8/site-packages/tvm/driver/tvmc/main.py", line 103, in _main
return args.func(args)
File "/home/kausthub/Desktop/major-project/env/lib/python3.8/site-packages/tvm/driver/tvmc/compiler.py", line 173, in drive_compile
compile_model(
File "/home/kausthub/Desktop/major-project/env/lib/python3.8/site-packages/tvm/driver/tvmc/compiler.py", line 337, in compile_model
graph_module = build(
File "/home/kausthub/Desktop/major-project/env/lib/python3.8/site-packages/tvm/driver/tvmc/compiler.py", line 410, in build
return relay.build(
File "/home/kausthub/Desktop/major-project/env/lib/python3.8/site-packages/tvm/relay/build_module.py", line 438, in build
graph_json, runtime_mod, params = bld_mod.build(
File "/home/kausthub/Desktop/major-project/env/lib/python3.8/site-packages/tvm/relay/build_module.py", line 161, in build
self._build(
File "tvm/_ffi/_cython/./packed_func.pxi", line 331, in tvm._ffi._cy3.core.PackedFuncBase.__call__
File "tvm/_ffi/_cython/./packed_func.pxi", line 276, in tvm._ffi._cy3.core.FuncCall
File "tvm/_ffi/_cython/./base.pxi", line 181, in tvm._ffi._cy3.core.CHECK_CALL
tvm._ffi.base.TVMError: Traceback (most recent call last):
32: TVMFuncCall
31: tvm::runtime::PackedFuncObj::Extractor<tvm::runtime::PackedFuncSubObj<tvm::relay::backend::RelayBuildModule::GetFunction(std::string const&, tvm::runtime::ObjectPtr<tvm::runtime::Object> const&)::{lambda(tvm::runtime::TVMArgs, tvm::runtime::TVMRetValue*)#3}> >::Call(tvm::runtime::PackedFuncObj const*, tvm::runtime::TVMArgs, tvm::runtime::TVMRetValue*)
30: tvm::relay::backend::RelayBuildModule::BuildRelay(tvm::IRModule, tvm::runtime::String const&)
29: tvm::relay::backend::RelayBuildModule::OptimizeImpl(tvm::IRModule)
28: tvm::transform::Pass::operator()(tvm::IRModule) const
27: tvm::transform::Pass::operator()(tvm::IRModule, tvm::transform::PassContext const&) const
26: tvm::transform::SequentialNode::operator()(tvm::IRModule, tvm::transform::PassContext const&) const
25: tvm::transform::Pass::operator()(tvm::IRModule, tvm::transform::PassContext const&) const
24: tvm::relay::transform::FunctionPassNode::operator()(tvm::IRModule, tvm::transform::PassContext const&) const
23: _ZN3tvm7runtime13PackedFuncObj9ExtractorINS0_16PackedFuncSubObjIZNS0_15TypedPackedFuncIFNS_5relay8FunctionES6_NS_8IRModuleENS_9transform11PassContextEEE17AssignTypedLambdaIZNS5_9transform13AlterOpLayoutEvEUlS6_S7_S9_E_EEvT_EUlRKNS0_7TVMArgsEPNS0_11TVMRetValueEE_EEE4CallEPKS1_SG_SK_
22: tvm::relay::alter_op_layout::AlterOpLayout(tvm::RelayExpr const&)
21: tvm::relay::ForwardRewrite(tvm::RelayExpr const&, tvm::runtime::TypedPackedFunc<tvm::RelayExpr (tvm::relay::Call const&, tvm::runtime::Array<tvm::RelayExpr, void> const&, tvm::runtime::ObjectRef const&)> const&, std::function<tvm::runtime::ObjectRef (tvm::relay::Call const&)>, std::function<tvm::RelayExpr (tvm::RelayExpr const&)>)
20: tvm::relay::MixedModeMutator::VisitExpr(tvm::RelayExpr const&)
19: tvm::relay::MixedModeMutator::VisitLeaf(tvm::RelayExpr const&)
18: _ZN3tvm5relay16M
17: tvm::relay::ExprMutator::VisitExpr(tvm::RelayExpr const&)
16: tvm::relay::ExprFunctor<tvm::RelayExpr (tvm::RelayExpr const&)>::VisitExpr(tvm::RelayExpr const&)
15: _ZZN3tvm5relay11ExprFuncto
14: tvm::relay::ExprMutator::VisitExpr_(tvm::relay::FunctionNode const*)
13: tvm::relay::MixedModeMutator::VisitExpr(tvm::RelayExpr const&)
12: tvm::relay::MixedModeMutator::VisitLeaf(tvm::RelayExpr const&)
11: _ZN3tvm5relay16M
10: tvm::relay::ExprMutator::VisitExpr(tvm::RelayExpr const&)
9: tvm::relay::ExprFunctor<tvm::RelayExpr (tvm::RelayExpr const&)>::VisitExpr(tvm::RelayExpr const&)
8: _ZZN3tvm5relay11ExprFuncto
7: tvm::relay::MixedModeMutator::VisitExpr_(tvm::relay::CallNode const*)
6: tvm::relay::ForwardRewriter::Rewrite_(tvm::relay::CallNode const*, tvm::RelayExpr const&)
5: tvm::runtime::TVMRetValue tvm::runtime::PackedFunc::operator()<tvm::relay::Call const&, tvm::runtime::Array<tvm::RelayExpr, void>&, tvm::runtime::ObjectRef>(tvm::relay::Call const&, tvm::runtime::Array<tvm::RelayExpr, void>&, tvm::runtime::ObjectRef&&) const
4: _ZN3tvm7runtime13PackedFun
3: tvm::runtime::TypedPackedFunc<tvm::RelayExpr (tvm::relay::Call const&, tvm::runtime::Array<tvm::RelayExpr, void> const&, tvm::runtime::ObjectRef const&)>::AssignTypedLambda<tvm::RelayExpr (*)(tvm::relay::Call const&, tvm::runtime::Array<tvm::RelayExpr, void> const&, tvm::runtime::ObjectRef const&)>(tvm::RelayExpr (*)(tvm::relay::Call const&, tvm::runtime::Array<tvm::RelayExpr, void> const&, tvm::runtime::ObjectRef const&))::{lambda(tvm::runtime::TVMArgs const&, tvm::runtime::TVMRetValue*)#1}::operator()(tvm::runtime::TVMArgs const&, tvm::runtime::TVMRetValue*) const
2: tvm::RelayExpr tvm::relay::LayoutRewriter<tvm::relay::alter_op_layout::AlterTransformMemorizer>(tvm::relay::Call const&, tvm::runtime::Array<tvm::RelayExpr, void> const&, tvm::runtime::ObjectRef const&)
1: tvm::relay::alter_op_layout::AlterTransformMemorizerNode::CallWithNewLayouts(tvm::relay::Call const&, tvm::Attrs, std::vector<tvm::RelayExpr, std::allocator<tvm::RelayExpr> > 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*) [clone .cold]
File "tvm/_ffi/_cython/./packed_func.pxi", line 56, in tvm._ffi._cy3.core.tvm_callback
File "/home/kausthub/Desktop/major-project/env/lib/python3.8/site-packages/tvm/relay/op/nn/_nn.py", line 232, in alter_op_layout_conv2d
return topi.nn.conv2d_alter_layout(attrs, inputs, tinfos, out_type)
File "/home/kausthub/Desktop/major-project/env/lib/python3.8/site-packages/decorator.py", line 232, in fun
return caller(func, *(extras + args), **kw)
File "/home/kausthub/Desktop/major-project/env/lib/python3.8/site-packages/tvm/target/generic_func.py", line 286, in dispatch_func
return dispatch_dict[k](*args, **kwargs)
File "/home/kausthub/Desktop/major-project/env/lib/python3.8/site-packages/tvm/topi/x86/conv2d_alter_op.py", line 60, in _alter_conv2d_layout
impl, outs = relay.backend.te_compiler.select_implementation(
File "/home/kausthub/Desktop/major-project/env/lib/python3.8/site-packages/tvm/relay/backend/te_compiler.py", line 206, in select_implementation
outs = impl.compute(attrs, inputs, out_type)
File "/home/kausthub/Desktop/major-project/env/lib/python3.8/site-packages/tvm/relay/op/op.py", line 126, in compute
return _OpImplementationCompute(self, attrs, inputs, out_type)
File "tvm/_ffi/_cython/./packed_func.pxi", line 331, in tvm._ffi._cy3.core.PackedFuncBase.__call__
File "tvm/_ffi/_cython/./packed_func.pxi", line 276, in tvm._ffi._cy3.core.FuncCall
File "tvm/_ffi/_cython/./base.pxi", line 181, in tvm._ffi._cy3.core.CHECK_CALL
3: TVMFuncCall
2: tvm::runtime::PackedFuncObj::Extractor<tvm::runtime::PackedFuncSubObj<tvm::relay::__mk_TVM6::{lambda(tvm::runtime::TVMArgs, tvm::runtime::TVMRetValue*)#1}> >::Call(tvm::runtime::PackedFuncObj const*, tvm::relay::__mk_TVM6, tvm::runtime::TVMRetValue)
1: tvm::relay::OpImplementation::Compute(tvm::Attrs const&, tvm::runtime::Array<tvm::te::Tensor, void> const&, tvm::Type 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*) [clone .cold]
File "tvm/_ffi/_cython/./packed_func.pxi", line 56, in tvm._ffi._cy3.core.tvm_callback
File "/home/kausthub/Desktop/major-project/env/lib/python3.8/site-packages/tvm/relay/op/strategy/generic.py", line 255, in _compute_conv2d
return [topi_compute(*args)]
File "/home/kausthub/Desktop/major-project/env/lib/python3.8/site-packages/tvm/topi/x86/conv2d.py", line 131, in conv2d_nchw
packed_out = conv2d_NCHWc(data, kernel, strides, padding, dilation, layout, layout, out_dtype)
File "/home/kausthub/Desktop/major-project/env/lib/python3.8/site-packages/tvm/autotvm/task/topi_integration.py", line 165, in wrapper
node = topi_compute(cfg, *args)
File "/home/kausthub/Desktop/major-project/env/lib/python3.8/site-packages/tvm/topi/x86/conv2d.py", line 196, in conv2d_NCHWc
cfg.define_split("tile_ic", in_channel, num_outputs=2)
File "/home/kausthub/Desktop/major-project/env/lib/python3.8/site-packages/tvm/autotvm/task/space.py", line 730, in define_split
return self._add_new_transform(SplitSpace, name, axes, policy, **kwargs)
File "/home/kausthub/Desktop/major-project/env/lib/python3.8/site-packages/tvm/autotvm/task/space.py", line 832, in _add_new_transform
axes = [x if isinstance(x, (VirtualAxis, Axis)) else self.axis(x) for x in axes]
File "/home/kausthub/Desktop/major-project/env/lib/python3.8/site-packages/tvm/autotvm/task/space.py", line 832, in <listcomp>
axes = [x if isinstance(x, (VirtualAxis, Axis)) else self.axis(x) for x in axes]
File "/home/kausthub/Desktop/major-project/env/lib/python3.8/site-packages/tvm/autotvm/task/space.py", line 687, in axis
return VirtualAxis(var)
File "/home/kausthub/Desktop/major-project/env/lib/python3.8/site-packages/tvm/autotvm/task/space.py", line 141, in __init__
raise RuntimeError("Invalid type of axis: " + str(type(var)))
RuntimeError: Invalid type of axis: <class 'tvm.tir.expr.Any'>