enhance cache write to support multiple tensors generated by ONE computeOp (#1042) (202570e4) · 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
202570e4

Commit 202570e4 authored 6 years ago by libing4752 Committed by Tianqi Chen 6 years ago

Downloads
- Patches
- Plain Diff

enhance cache write to support multiple tensors generated by ONE computeOp (#1042)

parent 239227d4

No related branches found

No related tags found

Loading

Changes 6

Hide whitespace changes

Inline Side-by-side

Showing

include/tvm/schedule.h 17 additions, 0 deletions

include/tvm/schedule.h
python/tvm/schedule.py 2 additions, 2 deletions

python/tvm/schedule.py
src/api/api_lang.cc 7 additions, 2 deletions

src/api/api_lang.cc
src/schedule/schedule_dataflow_rewrite.cc 73 additions, 20 deletions

src/schedule/schedule_dataflow_rewrite.cc
tests/python/integration/test_ewise.py 44 additions, 0 deletions

tests/python/integration/test_ewise.py
tests/python/unittest/test_schedule_schedule_ops.py 15 additions, 0 deletions

tests/python/unittest/test_schedule_schedule_ops.py

with 158 additions and 24 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