Skip to content
This repository was archived by the owner on Nov 17, 2023. It is now read-only.

Add support for boolean inputs to FusedOp#16796

Merged
ptrendx merged 7 commits into
apache:masterfrom
ptrendx:pr_fix_16723
Nov 23, 2019
Merged

Add support for boolean inputs to FusedOp#16796
ptrendx merged 7 commits into
apache:masterfrom
ptrendx:pr_fix_16723

Conversation

@ptrendx

@ptrendx ptrendx commented Nov 12, 2019

Copy link
Copy Markdown
Member

Description

Fixes #16723

This PR fixes 2 problems:

  • FusedOp did not support boolean input type as its inputs.
  • During testing the fix for the first problem another problem was spotted (which I believe was previously fixed during the process of working on the fusion PR, but I guess some merge undid it) - the FusedOp did not recompile the code when there was a change of inputs/outputs number of dimensions, which is wrong.

This PR fixes both of those problems and introduces tests for them.
@sxjscience Please validate the fix.

@ptrendx ptrendx requested a review from DickJC123 November 12, 2019 23:33
Comment thread tests/python/gpu/test_fusion.py Outdated
Comment thread tests/python/gpu/test_fusion.py Outdated
@ptrendx

ptrendx commented Nov 21, 2019

Copy link
Copy Markdown
Member Author

@marcoabreu @larroy Could you tell me what is the configuration of the unix-gpu test runners? They make TVM error out and I cannot reproduce it locally.

@ptrendx

ptrendx commented Nov 22, 2019

Copy link
Copy Markdown
Member Author

@sxjscience Could you validate the fix in your original code?

@sxjscience

Copy link
Copy Markdown
Member

@ptrendx I'm testing it locally

@sxjscience

Copy link
Copy Markdown
Member

@ptrendx Thanks for the fix. I've confirmed that it solves my previous failure.

@ptrendx ptrendx merged commit e3a7141 into apache:master Nov 23, 2019
ptrendx added a commit to ptrendx/mxnet that referenced this pull request Nov 23, 2019
* Support bool in fusion

* Added tests

* Use use_np decorator vs calling the numpy convention by hand

* Debug TVM failure

* Fix

* Do not invoke TVM broadcast kernel
@ptrendx ptrendx mentioned this pull request Nov 23, 2019
ptrendx added a commit that referenced this pull request Nov 23, 2019
* Support bool in fusion

* Added tests

* Use use_np decorator vs calling the numpy convention by hand

* Debug TVM failure

* Fix

* Do not invoke TVM broadcast kernel
@larroy

larroy commented Nov 27, 2019

Copy link
Copy Markdown
Contributor

@ptrendx is in the container in ci/ folder. You can use the exact same run by using ci/build.py you have the same instruction in the output of CI run either at the top or the bottom.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[Bug] fused_op does not support boolean type

3 participants