[SCHEDULE] Fix cross thread schedule after refactor (#85) (9ec40edd) · 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
9ec40edd

Commit 9ec40edd authored 7 years ago by Tianqi Chen Committed by GitHub 7 years ago

Downloads
- Patches
- Plain Diff

[SCHEDULE] Fix cross thread schedule after refactor (#85)

parent ea9c1c59

No related branches found

No related tags found

No related merge requests found

Changes 5

Hide whitespace changes

Inline Side-by-side

Showing

src/op/compute_op.cc 20 additions, 22 deletions

src/op/compute_op.cc
src/pass/lower_thread_allreduce.cc 12 additions, 8 deletions

src/pass/lower_thread_allreduce.cc
src/schedule/bound.cc 5 additions, 1 deletion

src/schedule/bound.cc
src/schedule/schedule_dataflow_rewrite.cc 10 additions, 0 deletions

src/schedule/schedule_dataflow_rewrite.cc
tests/python/integration/test_reduce.py 5 additions, 4 deletions

tests/python/integration/test_reduce.py

with 52 additions and 35 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