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

apache/tvm

main
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
14 branches 23 tags
Code

Latest commit

@Lunderberg
Lunderberg [UnitTest][NVPTX] Avoid cascading failures from CUDA postproc (#15136)
0bb390b Jul 5, 2023
[UnitTest][NVPTX] Avoid cascading failures from CUDA postproc (#15136) 
Prior to this commit, the tests in
`test_tir_transform_inject_ptx_async_copy.py` registered the
`"tvm_callback_cuda_postproc"` function during pytest collection, and
used a global variable to disable its functionality outside of the
tests in this file.  This had two major issues.  First, if any other
test also installs a postproc function, these postproc function
required by the NVPTX tests would be overwritten.  Second, if one of
the NTPTX tests fails, the global variable controlling the postproc
function would not be reset, causing any subsequent CUDA-related tests
to also fail.

This commit updates these NVPTX tests to conditionally install the
postproc function, to de-register it after the test instead of
disabling its functionality, and to de-register it regardless of the
test result.

This issue was initially found when debugging
#15103, when a failure in
`test_tir_transform_inject_ptx_async_copy.py::test_cp_async_in_if_then_else`
caused failures in 32 unrelated tests ([CI
link](https://ci.tlcpack.ai/blue/organizations/jenkins/tvm-gpu/detail/PR-15103/7/tests)).
0bb390b

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
.github
[Python] Enhance Wheel Packaging (#15167)
July 1, 2023 11:17
3rdparty
[BYOC] Add GEMM kernel from FasterTransformer as submodule (#15046)
June 10, 2023 00:48
apps
Update tvm_runtime.h (#15185)
July 1, 2023 22:36
ci
[Docker] Update docker images for llvm-16 (#15105)
June 16, 2023 09:44
cmake
[CMAKE] Add Vulkan header for Android (#15229)
July 5, 2023 08:41
conda
Update version to 0.14.dev0 on main branch (#15208)
July 3, 2023 12:40
configs
[TVMC] Add --config argument for config files (#11012)
April 19, 2022 11:50
docker
[Docker] Fix build.sh environment variables (#15149)
June 24, 2023 09:46
docs
[#15157][Rust][Doc] Re-enable the Rust documentation build (#15213)
July 4, 2023 23:23
gallery
[microNPU] Upgrade Vela to v3.8.0 (#15114)
June 23, 2023 16:33
golang
Fix typo in golang sample (#13476)
November 28, 2022 19:35
include/tvm
Update version to 0.14.dev0 on main branch (#15208)
July 3, 2023 12:40
jvm
[Target] Add target to all TVM callbacks (#14939)
May 25, 2023 09:10
licenses
[BYOC] Add GEMM kernel from FasterTransformer as submodule (#15046)
June 10, 2023 00:48
python
[Web] Increase default EMCC compilation total memory size (#15218)
July 4, 2023 08:29
rust
[CI] Modify test cases to accommodate the CI upgrades (#14651)
May 5, 2023 11:54
src
[ARITH] Enhance buffer shape bound deduction to include offset (#15228)
July 4, 2023 18:38
tests
[UnitTest][NVPTX] Avoid cascading failures from CUDA postproc (#15136)
July 4, 2023 23:24
vta
[CI] Modify test cases to accommodate the CI upgrades (#14651)
May 5, 2023 11:54
web
Update version to 0.14.dev0 on main branch (#15208)
July 3, 2023 12:40
.asf.yaml
[skip ci][ci][docker] Add cross compilation libs (#13800)
January 17, 2023 15:02
.clang-format
add .clang-format (#2395)
January 8, 2019 13:08
.gitattributes
[ci] Split Jenkinsfile into platform-specific jobs (#13300)
December 6, 2022 13:52
.gitignore
[tir] Add line level debug info (#13012)
January 6, 2023 16:03
.gitmodules
[BYOC] Add GEMM kernel from FasterTransformer as submodule (#15046)
June 10, 2023 00:48
.pre-commit-config.yaml
[CI] make pre-commit hooks to run on every push instead of every comm…
August 31, 2021 15:42
CMakeLists.txt
[BYOC] Add GEMM kernel from FasterTransformer as submodule (#15046)
June 10, 2023 00:48
CONTRIBUTORS.md
[COMMUNITY] Aleksei-grovety -> Reviewer (#15086)
June 13, 2023 09:19
KEYS
[COMMUNITY] Add new key for release signing (#14772)
May 5, 2023 18:53
LICENSE
More clear ref of thirdparty license (#14790)
May 6, 2023 16:47
Makefile
Fix make format (#11197)
May 2, 2022 15:58
NEWS.md
[ci] Lint for trailing newlines and spaces (#13058)
October 18, 2022 23:36
NOTICE
[#15043][Docs] Updated the copyright year from 2020 to 2023 (#15071)
June 17, 2023 10:13
README.md
[Tutorial] Fix formatting, grammar, dead link (#9281)
October 23, 2021 09:17
conftest.py
[CI] Deduplicate and clean XML test reports (#12332)
August 8, 2022 11:40
mypy.ini
[ci] Lint for trailing newlines and spaces (#13058)
October 18, 2022 23:36
pyproject.toml
[Diagnostics][Relay][InferType] Refactor InferType to work on whole m…
October 9, 2020 11:44
version.py
Update version to 0.14.dev0 on main branch (#15208)
July 3, 2023 12:40
Open Deep Learning Compiler Stack License Getting Started Contribute to TVM Acknowledgement

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

TVM is licensed under the Apache-2.0 license.

Getting Started

Check out the TVM Documentation site for installation instructions, tutorials, examples, and more. The Getting Started with TVM tutorial is a great place to start.

Contribute to TVM

TVM adopts apache committer model, we aim to create an open source project that is maintained and owned by the community. Check out 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

Code of conduct

Code of conduct

Security policy

Security policy

Stars

10k stars

Watchers

380 watching

Forks

3.1k forks
Report repository

Releases 14

Apache TVM v0.12.0 Latest
May 17, 2023
+ 13 releases

Packages

No packages published

Contributors 822

+ 811 contributors

Languages