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

Add support for boolean inputs to FusedOp #16796

Merged
merged 7 commits into from
Nov 23, 2019
Merged

Add support for boolean inputs to FusedOp #16796

Changes from 1 commit
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
d8c94ea
Support bool in fusion
ptrendx Nov 12, 2019
41e2a87
Added tests
ptrendx Nov 12, 2019
cbe1619
Merge branch 'upstream' into pr_fix_16723
ptrendx Nov 20, 2019
4e39f83
Use use_np decorator vs calling the numpy convention by hand
ptrendx Nov 20, 2019
6e62840
Debug TVM failure
ptrendx Nov 21, 2019
ebf65fe
Fix
ptrendx Nov 21, 2019
f9982d8
Do not invoke TVM broadcast kernel
ptrendx Nov 21, 2019
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
4 changes: 2 additions & 2 deletions tests/python/gpu/test_fusion.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -239,9 +239,9 @@ def test_fusion_compiler_cache():
check_fused_symbol(a+b, ctx=mx.gpu(1), a=arr1, b=arr2)

@with_seed()
@use_np
def test_fusion_boolean_inputs():
from mxnet.gluon import HybridBlock
mx.npx.set_np()

class Foo(HybridBlock):
def __init__(self, prefix=None, params=None):
Expand All @@ -256,7 +256,7 @@ def hybrid_forward(self, F, valid_length):
foo = Foo()
foo.hybridize(static_alloc=True)
out = foo(mx.np.ones((10,), ctx=mx.gpu()))
mx.npx.reset_np()
mx.nd.waitall()
ptrendx marked this conversation as resolved.
Show resolved Hide resolved

@with_seed()
def test_fusion_different_dimensions():
Expand Down