Skip to content

Commit

Permalink
Fix typo in travserse (apache#4469)
Browse files Browse the repository at this point in the history
  • Loading branch information
apivovarov authored and Xingyu Zhou committed Dec 13, 2019
1 parent d6f6f43 commit 3049d84
Show file tree
Hide file tree
Showing 18 changed files with 28 additions and 28 deletions.
2 changes: 1 addition & 1 deletion python/tvm/intrin.py
Original file line number Diff line number Diff line change
Expand Up @@ -575,7 +575,7 @@ def register_intrin_rule(target, intrin, f=None, override=False):
The name of codegen target.
intrin : str
The name of the instrinsic.
The name of the intrinsic.
f : function, optional
The function to be registered.
Expand Down
2 changes: 1 addition & 1 deletion src/api/api_pass.cc
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
*/

/*!
* Exposre of pass functions.
* Exposure of pass functions.
* \file api_pass.cc
*/
#include <tvm/expr.h>
Expand Down
4 changes: 2 additions & 2 deletions src/pass/lower_warp_memory.cc
Original file line number Diff line number Diff line change
Expand Up @@ -46,9 +46,9 @@ namespace ir {
// This requires us to do the following rewriting:
// - Rewrite allocation to use local memory.
// - Rewrite store of warp memory to local store.
// - Rewrite load of waro memory to local plus a shuffle.
// - Rewrite load of warp memory to local plus a shuffle.
//
// Define a generic shuffle instrinsic warp_shuffle(data, warp_index).
// Define a generic shuffle intrinsic warp_shuffle(data, warp_index).
// We can use the following rewriting rule
//
// Before rewrite,
Expand Down
2 changes: 1 addition & 1 deletion topi/python/topi/arm_cpu/bitserial_dense.py
Original file line number Diff line number Diff line change
Expand Up @@ -158,7 +158,7 @@ def _schedule(cfg, s, data_vec, weight_vec, output, unipolar):
return s

def traverse(op):
"""Internal travserse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
if tag.is_broadcast(op.tag) or 'elemwise' in op.tag:
if op not in s.outputs:
Expand Down
2 changes: 1 addition & 1 deletion topi/python/topi/bifrost/depthwise_conv2d.py
Original file line number Diff line number Diff line change
Expand Up @@ -104,7 +104,7 @@ def tile_and_bind3d(tensor, z, y, x, z_factor=2, y_factor=None, x_factor=None):
s[conv].compute_at(s[output], ji)

def traverse(op):
"""Internal travserse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
if tag.is_broadcast(op.tag):
if op not in s.outputs:
Expand Down
2 changes: 1 addition & 1 deletion topi/python/topi/cuda/dense.py
Original file line number Diff line number Diff line change
Expand Up @@ -105,7 +105,7 @@ def _schedule(C):
scheduled_ops = []

def traverse(OP):
"""Internal travserse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
if tag.is_broadcast(OP.tag):
if OP not in s.outputs:
Expand Down
2 changes: 1 addition & 1 deletion topi/python/topi/cuda/depthwise_conv2d.py
Original file line number Diff line number Diff line change
Expand Up @@ -192,7 +192,7 @@ def _schedule(temp, Filter, DepthwiseConv2d):
scheduled_ops = []

def traverse(OP):
"""Internal travserse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
if tag.is_broadcast(OP.tag):
if OP not in s.outputs:
Expand Down
4 changes: 2 additions & 2 deletions topi/python/topi/cuda/pooling.py
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,7 @@ def _schedule(Pool):
scheduled_ops = []

def traverse(OP):
"""Internal travserse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
if tag.is_broadcast(OP.tag):
if OP not in s.outputs:
Expand Down Expand Up @@ -131,7 +131,7 @@ def _schedule(PaddedInput, Pool):
scheduled_ops = []

def traverse(OP):
"""Internal travserse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
if tag.is_broadcast(OP.tag):
if OP not in s.outputs:
Expand Down
4 changes: 2 additions & 2 deletions topi/python/topi/cuda/reduction.py
Original file line number Diff line number Diff line change
Expand Up @@ -109,7 +109,7 @@ def schedule_reduce(outs):
scheduled_ops = []

def traverse_before_reduce(operator):
"""Internal travserse function"""
"""Internal traverse function"""
if isinstance(operator, tvm.tensor.PlaceholderOp):
return
if tag.is_injective(operator.tag):
Expand All @@ -123,7 +123,7 @@ def traverse_before_reduce(operator):
scheduled_ops.append(operator)

def traverse_after_reduce(operator):
"""Internal travserse function"""
"""Internal traverse function"""
if tag.is_broadcast(operator.tag):
if operator not in scheduled_ops:
schedule_injective_from_existing(sch, operator.output(0))
Expand Down
10 changes: 5 additions & 5 deletions topi/python/topi/hls/nn.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ def _schedule_conv2d(outs):
tvm.schedule.AutoInlineInjective(s)

def traverse(OP):
"""Internal travserse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
if tag.is_injective(OP.tag):
if OP not in s.outputs:
Expand Down Expand Up @@ -214,7 +214,7 @@ def schedule_reduce(outs):
tvm.schedule.AutoInlineInjective(s)

def traverse(OP):
"""Internal travserse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
if tag.is_broadcast(OP.tag):
if OP not in s.outputs:
Expand Down Expand Up @@ -306,7 +306,7 @@ def schedule_dense(outs):
tvm.schedule.AutoInlineInjective(s)

def traverse(OP):
"""Internal travserse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
if tag.is_broadcast(OP.tag):
if OP not in s.outputs:
Expand Down Expand Up @@ -350,7 +350,7 @@ def schedule_pool(outs, layout):
tvm.schedule.AutoInlineInjective(s)

def traverse(OP):
"""Internal travserse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
if tag.is_broadcast(OP.tag):
if OP not in s.outputs:
Expand Down Expand Up @@ -394,7 +394,7 @@ def schedule_adaptive_pool(outs):
tvm.schedule.AutoInlineInjective(s)

def traverse(OP):
"""Internal travserse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
if tag.is_broadcast(OP.tag):
if OP not in s.outputs:
Expand Down
2 changes: 1 addition & 1 deletion topi/python/topi/intel_graphics/depthwise_conv2d.py
Original file line number Diff line number Diff line change
Expand Up @@ -193,7 +193,7 @@ def _schedule(temp, Filter, DepthwiseConv2d):
scheduled_ops = []

def traverse(OP):
"""Internal travserse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
if tag.is_broadcast(OP.tag):
if OP not in s.outputs:
Expand Down
2 changes: 1 addition & 1 deletion topi/python/topi/opengl/conv2d_nchw.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,7 @@ def _schedule(conv2d, data):
s[data].opengl()

def traverse(OP):
"""Internal travserse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
if tag.is_broadcast(OP.tag):
if OP not in s.outputs:
Expand Down
2 changes: 1 addition & 1 deletion topi/python/topi/opengl/dense.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,7 @@ def _schedule(Dense):
s[Out].opengl()

def traverse(OP):
"""Internal travserse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
if tag.is_broadcast(OP.tag):
if OP not in s.outputs:
Expand Down
4 changes: 2 additions & 2 deletions topi/python/topi/opengl/pooling.py
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,7 @@ def _schedule(Pool):
s[Out].opengl()

def traverse(OP):
"""Internal travserse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
if tag.is_broadcast(OP.tag):
if OP not in s.outputs:
Expand Down Expand Up @@ -102,7 +102,7 @@ def _schedule(PaddedInput, Pool):
s[Out].opengl()

def traverse(OP):
"""Internal travserse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
if tag.is_broadcast(OP.tag):
if OP not in s.outputs:
Expand Down
2 changes: 1 addition & 1 deletion topi/python/topi/x86/binary_dense.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ def _schedule(A, B, C):
s[Out].vectorize(xi)

def traverse(OP):
"""Internal travserse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
if tag.is_broadcast(OP.tag):
if OP not in s.outputs:
Expand Down
2 changes: 1 addition & 1 deletion topi/python/topi/x86/bitserial_dense.py
Original file line number Diff line number Diff line change
Expand Up @@ -69,7 +69,7 @@ def _schedule(cfg, s, data_vec, weight_vec, output):
return s

def traverse(op):
"""Internal travserse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
if tag.is_broadcast(op.tag) or 'elemwise' in op.tag:
if op not in s.outputs:
Expand Down
4 changes: 2 additions & 2 deletions topi/python/topi/x86/pooling.py
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,7 @@ def _schedule(PaddedInput, Pool):
_parallel_sch(s[Pool], outs[0].shape, do_vectorize)

def traverse(OP):
"""Internal travserse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
if tag.is_broadcast(OP.tag):
if OP not in s.outputs:
Expand Down Expand Up @@ -137,7 +137,7 @@ def schedule_adaptive_pool(outs):
scheduled_ops = []

def traverse(OP):
"""Internal travserse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
if tag.is_broadcast(OP.tag):
if OP not in s.outputs:
Expand Down
4 changes: 2 additions & 2 deletions topi/python/topi/x86/reduction.py
Original file line number Diff line number Diff line change
Expand Up @@ -78,7 +78,7 @@ def schedule_reduce(outs):
scheduled_ops = []

def traverse_before_reduce(operator):
"""Internal travserse function"""
"""Internal traverse function"""
if isinstance(operator, tvm.tensor.PlaceholderOp):
return
if tag.is_injective(operator.tag):
Expand All @@ -92,7 +92,7 @@ def traverse_before_reduce(operator):
scheduled_ops.append(operator)

def traverse_after_reduce(operator):
"""Internal travserse function"""
"""Internal traverse function"""
if tag.is_broadcast(operator.tag):
if operator not in scheduled_ops:
generic.schedule_injective_from_existing(sch, operator)
Expand Down

0 comments on commit 3049d84

Please sign in to comment.