Commit f9c17997 by Tatsuya Nishiyama Committed by Tianqi Chen

Fix tutorial to follow a change of elemwise_sum (#1374)

parent 5e018774
...@@ -65,7 +65,7 @@ d = a * b # same as topi.broadcast_mul ...@@ -65,7 +65,7 @@ d = a * b # same as topi.broadcast_mul
# we can schedule the following series of operations ending with :code:`topi.sum` using only # we can schedule the following series of operations ending with :code:`topi.sum` using only
# :code:`topi.generic.schedule_reduce` # :code:`topi.generic.schedule_reduce`
# #
e = topi.elemwise_sum([c, d], num_args=2) e = topi.elemwise_sum([c, d])
f = e / 2.0 f = e / 2.0
g = topi.sum(f) g = topi.sum(f)
with tvm.target.cuda(): with tvm.target.cuda():
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment