[PASS] Improve vthread injection. (#411) (5072efae) · Commits · cld / ml / tvm · GitLab

Skip to content

GitLab

Explore

Sign in

Primary navigation

Project

T

tvm
- Activity
- Members
- Labels
- Issues
- Issue boards
- Milestones
- Wiki
- Releases
- Model registry
- Environments
- Incidents
- Service Desk

Snippets Groups Projects

- cld
- ml
- tvm
- Commits
5072efae

Commit 5072efae authored 7 years ago by Tianqi Chen Committed by GitHub 7 years ago

Downloads
- Patches
- Plain Diff

[PASS] Improve vthread injection. (#411)

parent b0d9f299

No related branches found

No related tags found

No related merge requests found

Changes 6

Hide whitespace changes

Inline Side-by-side

Showing

include/tvm/ir.h 5 additions, 0 deletions

include/tvm/ir.h
src/op/op_util.cc 2 additions, 1 deletion

src/op/op_util.cc
src/pass/inject_double_buffer.cc 6 additions, 6 deletions

src/pass/inject_double_buffer.cc
src/pass/inject_virtual_thread.cc 64 additions, 23 deletions

src/pass/inject_virtual_thread.cc
src/pass/lower_tvm_builtin.cc 2 additions, 0 deletions

src/pass/lower_tvm_builtin.cc
tests/python/unittest/test_pass_inject_vthread.py 32 additions, 0 deletions

tests/python/unittest/test_pass_inject_vthread.py

with 111 additions and 30 deletions

Loading

0% Loading or .

You are about to add 0 people to the discussion. Proceed with caution.

Finish editing this message first!

Please register or sign in to comment