Skip to content

[BYOC] Refine AnnotateTarget and MergeCompilerRegion Passes #5277

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
zhiics merged 17 commits into from
Apr 10, 2020
Merged

[BYOC] Refine AnnotateTarget and MergeCompilerRegion Passes #5277

Changes from 1 commit
Commits
Show all changes
17 commits
Select commit Hold shift + click to select a range
333b215
add target to region
comaniac Apr 3, 2020
36d20da
refactor annotate_target
comaniac Apr 3, 2020
ac040a0
Make all unit test working
comaniac Apr 3, 2020
10add0d
quick fix
comaniac Apr 3, 2020
b89d8c2
enable BN, unit test failed
comaniac Apr 3, 2020
0bd1128
quick fix fusion
comaniac Apr 4, 2020
e8a41da
Fix vm test, unit test. Refactor annotate_target a bit.
zhiics Apr 3, 2020
5c9adff
revert fusion change
zhiics Apr 4, 2020
8e3e014
style fix
zhiics Apr 4, 2020
cfb7608
Refactor merge region pass
comaniac Apr 6, 2020
9e869c4
format
comaniac Apr 6, 2020
4515403
minor fix
zhiics Apr 7, 2020
41147a9
Skip e2e test
comaniac Apr 7, 2020
4a7bf3a
lint
comaniac Apr 8, 2020
3da42ed
support AnnotateTarget multiple runs
comaniac Apr 8, 2020
e2b7a7e
Add HasAttr and revert DNNL codegen
comaniac Apr 9, 2020
13cfa83
address comment
comaniac Apr 9, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
quick fix
  • Loading branch information
@comaniac
comaniac committed Apr 10, 2020
commit 10add0d357b317b40046b4eb0e520c1b5bbe4b1e
4 changes: 2 additions & 2 deletions tests/python/relay/test_pass_annotate_target.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -229,7 +229,7 @@ def after():
def test_type_propagation():
target = "test_type_propagation"

@reg.register("nn.relu", "target.test_type_propagation" + target)
@reg.register("nn.relu", "target." + target)
def relu(attrs, args): # pylint: disable=unused-variable
return args[0].checked_type.dtype == "float32"

Expand Down Expand Up @@ -281,7 +281,7 @@ def after():
tup = relay.Tuple([cb_3, cb_4])
ce_3 = relay.annotation.compiler_end(tup, target)
cb_3 = relay.annotation.compiler_begin(ce_3, target)
out = relay.op.concatenate(cb_3, 1)
out = relay.op._make.concatenate(cb_3, 1)
ce_4 = relay.annotation.compiler_end(out, target)
f = relay.Function([x, y], ce_4)
mod = tvm.IRModule.from_expr(f)
Expand Down