Skip to content

Commit

Permalink
Send list as argument to schedule_conv2d (apache#4358)
Browse files Browse the repository at this point in the history
When getting cuda schedule passing single tensor seem to work but after changing target to "llvm" causes assert.
Sending list on other hand makes both cuda and llvm targets happy.
See https://discuss.tvm.ai/t/solved-simple-example-error-attributeerror-tensorslice-object-has-no-attribute-op/2245/3
  • Loading branch information
vmiheer authored and Xingyu Zhou committed Nov 26, 2019
1 parent 522c6f9 commit d909e0a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tutorials/topi/intro_topi.py
Original file line number Diff line number Diff line change
Expand Up @@ -135,7 +135,7 @@
with tvm.target.create("cuda"):
conv = topi.nn.conv2d(data, kernel, strides=1, padding=2, dilation=1)
out = topi.nn.relu(conv)
sconv = topi.generic.nn.schedule_conv2d_nchw(out)
sconv = topi.generic.nn.schedule_conv2d_nchw([out])
print(tvm.lower(sconv, [data, kernel], simple_mode=True))

######################################################################
Expand Down

0 comments on commit d909e0a

Please sign in to comment.