Skip to content
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.

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

Improve NNVM to Relay conversion #2734

Merged
merged 3 commits into from
Mar 8, 2019
Merged

Conversation

kazum
Copy link
Contributor

@kazum kazum commented Mar 5, 2019

This PR adds to_relay() checks to our NNVM tests. This PR also includes some cleanups of to_relay, fixes of detected bugs, and supports of some operators to pass the test.

@jroesch @yzhliu @grwlf please help review the PR.

@@ -441,6 +443,23 @@ def check_function(symbol, forward=None, backward=None, grad_input_vars=None,
debug_stage = "running"
nnvm_res = main_function(**np_inputs)

try:
logging.debug("checking to_relay conversion")
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Will this print every time this code is executed?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think it depends on the logging configuration, but, with our CI, this message will be printed only when an error occurs.

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

ok, seems ok

@jroesch
Copy link
Member

jroesch commented Mar 8, 2019

Overall looks good, I could swear their were more tests written by me, but I think were moved/removed during someone's overhaul of the tests.

@jroesch jroesch merged commit 695647d into apache:master Mar 8, 2019
@kazum kazum deleted the improve_to_relay branch March 9, 2019 21:02
wweic pushed a commit to neo-ai/tvm that referenced this pull request Mar 9, 2019
* Improve NNVM to Relay conversion

* fix pylint

* support __lshift_scalar__, abs, ceil, floor, and trunc to pass CI
wweic pushed a commit to neo-ai/tvm that referenced this pull request Mar 12, 2019
* Improve NNVM to Relay conversion

* fix pylint

* support __lshift_scalar__, abs, ceil, floor, and trunc to pass CI
wweic pushed a commit to neo-ai/tvm that referenced this pull request Mar 12, 2019
* Improve NNVM to Relay conversion

* fix pylint

* support __lshift_scalar__, abs, ceil, floor, and trunc to pass CI
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants