When I run matrix multiplication with size 1kX1kX1k the training runs.
With sizes 2kX2kX2k thought it consistently fails with error:
result: MeasureResult(costs=(RuntimeError('Traceback (most recent call last):\n [bt] (3) /home/vaidya.56/ml/ctvm/lib/python3.7/site-packages/tvm-0.6.dev0-py3.7-linux-x86_64.egg/tvm/libtvm.so(TVMFuncCall+0x48) [0x14944c70a9f8]\n [bt] (2) /home/vaidya.56/ml/ctvm/lib/python3.7/site-packages/tvm-0.6.dev0-py3.7-linux-x86_64.egg/tvm/libtvm.so(std::_Function_handler<void (tvm::runtime::TVMArgs, tvm::runtime::TVMRetValue*), tvm::runtime::RPCModuleNode::WrapRemote(void*)::{lambda(tvm::runtime::TVMArgs, tvm::runtime::TVMRetValue*)#1}>::_M_invoke(std::_Any_data const&, tvm::runtime::TVMArgs&&, tvm::runtime::TVMRetValue*&&)+0x3b) [0x14944c76e45b]\n [bt] (1) /home/vaidya.56/ml/ctvm/lib/python3.7/site-packages/tvm-0.6.dev0-py3.7-linux-x86_64.egg/tvm/libtvm.so(tvm::runtime::RPCSession::CallFunc(void*, tvm::runtime::TVMArgs, tvm::runtime::TVMRetValue*, tvm::runtime::PackedFunc const*)+0x175) [0x14944c776845]\n [bt] (0) /home/vaidya.56/ml/ctvm/lib/python3.7/site-packages/tvm-0.6.dev0-py3.7-linux-x86_64.egg/tvm/libtvm.so(dmlc::LogMessageFatal::~LogMessageFatal()+0x22) '),), error_no=4, all_cost=10.230645656585693, timestamp=1574050733.5890949)
I am able to run 2kX2kX2k c++ matmult though.