tvm
tvm copied to clipboard
TVM integration into PyTorch
Environments: Ubuntu16.04 LTS, Python 3.6.9, LLVM 8.0, GCC 5.4, Pytorch Nightly version (build from source), torch_tvm master branch. Successfully import tvm, but encounter error when import torch_tvm:  Bottom of...
python setup.py test output: ========================================================================================================= test session starts ========================================================================================================= platform linux -- Python 3.7.4, pytest-5.3.2, py-1.8.1, pluggy-0.13.1 rootdir: /tvm, inifile: setup.cfg, testpaths: test collected 0 items / 3 errors ===============================================================================================================...
Detect the C++ ABI of LLVM and PyTorch libs, ensure they are compatible and then build torch_tvm accordingly.
Hi, I have two questions: - How to save the compiled graph? Does saving the compiled graph save the tvm::CompilationGroup symbols along with the compiled subgraphs? I hope it will...
`python3 test/benchmarks.py` output: ```Tracing model with JIT Warming JIT up with 10 runs Running JIT 100 times Done benchmarking JIT Tracing model with TVM Warming TVM up with 10 iters...
It seems that there is no 3D convolution support. the tests are 1D and 2D. Is it correct? if so, are there plan to add 3D support? 3D convolution is...
Usage examples, comparations using tvm or vanilla pytorch?
I have installed the torch nightly build (torch-1.3.0.dev20190819-cp36-cp36m-linux_x86_64.whl) and successfully build torch_tvm. However, when I try to import torch_tvm I get the following error: ``` >>> import torch_tvm Traceback (most...
I have installed torch_tvm with cuda/opencl support by enabling the following options: https://github.com/dmlc/tvm/blob/master/cmake/config.cmake#L32 https://github.com/dmlc/tvm/blob/master/cmake/config.cmake#L129 https://github.com/dmlc/tvm/blob/master/cmake/config.cmake#L132 Trying to compare the cpu vs gpu performance by running the following test: https://github.com/pytorch/tvm/blob/master/test/benchmarks.py -...
I was just playing around with this optimization pass with the main PyTorch project. I was curious to understand why [control flow operators are not handled here](https://github.com/pytorch/tvm/blob/master/torch_tvm/fusion_pass.cpp#L30). What was the...