[BYOC] Refine AnnotateTarget and MergeCompilerRegion Passes (#5277)
authorCody Yu <comaniac0422@gmail.com>
Fri, 10 Apr 2020 21:32:56 +0000 (14:32 -0700)
committerGitHub <noreply@github.com>
Fri, 10 Apr 2020 21:32:56 +0000 (14:32 -0700)
commitf506c8b19ab3a7634ac56f63298dade6d40d2d1d
treec2c1d4bb96349ffdc923383fe6d085c28449f707
parent5795539c0d15a94c2278b576b857a48a40bae824
[BYOC] Refine AnnotateTarget and MergeCompilerRegion Passes (#5277)

* add target to region

* refactor annotate_target

* Make all unit test working

* quick fix

* enable BN, unit test failed

* Fix vm test, unit test. Refactor annotate_target a bit.

* quick fix fusion

* revert fusion change

* style fix

* Refactor merge region pass

* format

* minor fix

* Skip e2e test

* lint

* support AnnotateTarget multiple runs

* Add HasAttr and revert DNNL codegen

* address comment

Co-authored-by: Zhi Chen <chzhi@amazon.com>
15 files changed:
python/tvm/relay/op/contrib/dnnl.py
python/tvm/relay/transform/transform.py
src/relay/analysis/annotated_region_set.cc
src/relay/analysis/annotated_region_set.h
src/relay/backend/contrib/dnnl/codegen.cc
src/relay/backend/vm/compiler.cc
src/relay/transforms/annotate_target.cc
src/relay/transforms/merge_compiler_regions.cc
src/relay/transforms/partition_graph.cc
src/runtime/contrib/dnnl/dnnl.cc
src/runtime/contrib/dnnl/dnnl_kernel.h
tests/python/relay/test_annotated_regions.py
tests/python/relay/test_pass_annotate_target.py [moved from tests/python/relay/test_annotate_target.py with 72% similarity]
tests/python/relay/test_pass_merge_compiler_regions.py
tests/python/relay/test_pass_partition_graph.py