tvm icon indicating copy to clipboard operation
tvm copied to clipboard

TVM integration into PyTorch

Results 30 tvm issues
Sort by recently updated
recently updated
newest added

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: ![LLVM-error](https://user-images.githubusercontent.com/16394892/68832881-050d6080-06ed-11ea-85a7-5ec8a55b8660.png) 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.

enhancement

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...

bug

`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...

bug

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?

enhancement
help wanted

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...