The Wayback Machine - https://web.archive.org/web/20210226044603/https://github.com/apache/tvm
Skip to content

/ tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators

tvm.apache.org/
Apache-2.0 License
6.3k stars 1.9k forks
Star
Notifications
main
Switch branches/tags
5 branches 8 tags
Go to file
Code
Clone

Use Git or checkout with SVN using the web URL.

Latest commit

@ymwangg Yanming Wang
ymwangg and Yanming Wang [Frontend][Tensorflow] Add unique operator (#7441)
09b0c8e Feb 26, 2021
[Frontend][Tensorflow] Add unique operator (#7441) 
* Initial commit of the unique operator

Add unit tests for unique operator

* Add tensorflow unique op

* Refactor unique to use sort-based algorithm

* Change relay.unique test to run only on cpu

* Change topi.unique test to run only on cpu

* Change range to parallel for parallelizable loops

* Add return_counts option for relay.unique and topi.unique, add pytorch frontend

* Fix pylint

* Patch pytorch frontend

* Initial support of topi.cuda.unique

* Refactor to use ir_builder directly

* Modularize adjacent difference

* Refactor to simplify

* Fix typo

* Combine _unique and _unique_with_counts

* Reuse indices_ptr to remove arange_ptr

Co-authored-by: Yanming Wang <yanmwang@amazon.com>
09b0c8e

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
.github
[CI] Update actions miniconda (#6926)
Nov 16, 2020
3rdparty
update vta-hw version (#7271)
Jan 14, 2021
apps
Many fixes to get unit tests passing on Windows. (#7431)
Feb 25, 2021
cmake
Many fixes to get unit tests passing on Windows. (#7431)
Feb 25, 2021
conda
Many fixes to get unit tests passing on Windows. (#7431)
Feb 25, 2021
docker
[BYOC][VitisAI] Fix issue in Vitis AI codegen out tensor names matchi…
Feb 24, 2021
docs
[BYOC][VitisAI] Fix issue in Vitis AI codegen out tensor names matchi…
Feb 24, 2021
golang
[REFACTOR] Remainings of util => utils (#6778)
Oct 29, 2020
include/tvm
[Frontend][Tensorflow] Add unique operator (#7441)
Feb 26, 2021
jvm
[DOCS] Update to reflect the repo name change (#6967)
Nov 24, 2020
licenses
Bring Your Own Datatypes (#5812)
Sep 26, 2020
nnvm
Fixed minor misspelling (#7499)
Feb 22, 2021
python
[Frontend][Tensorflow] Add unique operator (#7441)
Feb 26, 2021
rust
[Rust] More Rust bindings for Attrs (#7082)
Dec 26, 2020
src
[Frontend][Tensorflow] Add unique operator (#7441)
Feb 26, 2021
tests
[Frontend][Tensorflow] Add unique operator (#7441)
Feb 26, 2021
tutorials
[CUDA][THRUST] Enforce -libs=thrust to allow thrust offload (#7468)
Feb 22, 2021
vta
Introduce module_loader to AutoTVM. (#7337)
Feb 25, 2021
web
[DOCS] Update to reflect the repo name change (#6967)
Nov 24, 2020
.asf.yaml
[INFO] Add .asf.yaml for github info (#4761)
Jan 21, 2020
.clang-format
add .clang-format (#2395)
Jan 8, 2019
.gitignore
[CONDA] Revamp conda recipe. (#6732)
Oct 24, 2020
.gitmodules
[VTA] HW sources refactor (#5188)
Mar 31, 2020
CMakeLists.txt
Many fixes to get unit tests passing on Windows. (#7431)
Feb 25, 2021
CONTRIBUTORS.md
[COMMUNITY] @d-smirnov -> reviewer (#7510)
Feb 24, 2021
Jenkinsfile
Report JUnit test results for all TVM Python tests (#7450)
Feb 17, 2021
KEYS
Zhi's key for ASF release (#6554)
Sep 26, 2020
LICENSE
Bring Your Own Datatypes (#5812)
Sep 26, 2020
Makefile
[CI] Move to use main as the default (#6665)
Oct 12, 2020
NEWS.md
[CI] Move to use main as the default (#6665)
Oct 12, 2020
NOTICE
[DOCS] Remove incubating from docs (#7525)
Feb 25, 2021
README.md
[DOCS] Remove incubating from docs (#7525)
Feb 25, 2021
conftest.py
[Format] Convert all Python code w/o CI (#6448)
Sep 11, 2020
pyproject.toml
[Diagnostics][Relay][InferType] Refactor InferType to work on whole m…
Oct 9, 2020
pytest.ini
[TESTS] Refactor tests to run on either the GPU or CPU. (#6331)
Sep 2, 2020
version.py
[REFACTOR] Remainings of util => utils (#6778)
Oct 29, 2020

README.md

Open Deep Learning Compiler Stack

Documentation | Contributors | Community | Release Notes

Build Status WinMacBuild

Apache TVM is a compiler stack for deep learning systems. It is designed to close the gap between the productivity-focused deep learning frameworks, and the performance- and efficiency-focused hardware backends. TVM works with deep learning frameworks to provide end to end compilation to different backends.

License

© Contributors Licensed under an Apache-2.0 license.

Contribute to TVM

TVM adopts apache committer model, we aim to create an open source project that is maintained and owned by the community. Checkout the Contributor Guide

Acknowledgement

We learned a lot from the following projects when building TVM.

  • Halide: Part of TVM's TIR and arithmetic simplification module originates from Halide. We also learned and adapted some part of lowering pipeline from Halide.
  • Loopy: use of integer set analysis and its loop transformation primitives.
  • Theano: the design inspiration of symbolic scan operator for recurrence.

About

Open deep learning compiler stack for cpu, gpu and specialized accelerators

tvm.apache.org/

Topics

javascript machine-learning performance deep-learning metal compiler gpu vulkan opencl tensor spirv rocm tvm

Resources

Readme

License

Apache-2.0 License

Releases 8

Apache TVM (incubating) v0.7.0 Latest
Oct 2, 2020
+ 7 releases

Packages

No packages published

Contributors 484

+ 473 contributors

Languages