Skip to content

Commit

Permalink
formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
tkonolige committed Oct 30, 2020
1 parent 284b6e9 commit 879481c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/tvm/topi/x86/nn.py
Original file line number Diff line number Diff line change
Expand Up @@ -87,7 +87,7 @@ def schedule_embed_grad(outs):
"""
s = te.create_schedule([outs[0].op])

vec_size = 8 # should autotune this, but we can't with hybrid script
vec_size = 8 # should autotune this, but we can't with hybrid script
out = s.outputs[0].output(0)
zi, zj, i, j = s[out].op.axis
zjo, zji = s[out].split(zj, factor=vec_size)
Expand Down

0 comments on commit 879481c

Please sign in to comment.