[PASS] Remap thread axis. (#1122) (154104b3) · 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
154104b3

Commit 154104b3 authored 6 years ago by Tianqi Chen Committed by GitHub 6 years ago

Downloads
- Patches
- Plain Diff

[PASS] Remap thread axis. (#1122)

parent d0f40112

No related branches found

No related tags found

Loading

Changes 8

Hide whitespace changes

Inline Side-by-side

Showing

include/tvm/build_module.h 6 additions, 0 deletions

include/tvm/build_module.h
include/tvm/ir_pass.h 14 additions, 0 deletions

include/tvm/ir_pass.h
python/tvm/build_module.py 8 additions, 0 deletions

python/tvm/build_module.py
python/tvm/target.py 1 addition, 0 deletions

python/tvm/target.py
src/api/api_pass.cc 1 addition, 0 deletions

src/api/api_pass.cc
src/codegen/build_module.cc 2 additions, 0 deletions

src/codegen/build_module.cc
src/pass/remap_thread_axis.cc 83 additions, 0 deletions

src/pass/remap_thread_axis.cc
tests/python/integration/test_ewise.py 3 additions, 2 deletions

tests/python/integration/test_ewise.py

with 118 additions and 2 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