Skip to content
Projects
Groups
Snippets
Help
This project
Loading...
Sign in / Register
Toggle navigation
T
tic
Overview
Overview
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
wenyuanbo
tic
Commits
7cf1eadc
Commit
7cf1eadc
authored
Dec 06, 2019
by
Alexander Pivovarov
Committed by
Wuwei Lin
Dec 06, 2019
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Fix typo in travserse (#4469)
parent
ba9d96bc
Hide whitespace changes
Inline
Side-by-side
Showing
18 changed files
with
28 additions
and
28 deletions
+28
-28
python/tvm/intrin.py
+1
-1
src/api/api_pass.cc
+1
-1
src/pass/lower_warp_memory.cc
+2
-2
topi/python/topi/arm_cpu/bitserial_dense.py
+1
-1
topi/python/topi/bifrost/depthwise_conv2d.py
+1
-1
topi/python/topi/cuda/dense.py
+1
-1
topi/python/topi/cuda/depthwise_conv2d.py
+1
-1
topi/python/topi/cuda/pooling.py
+2
-2
topi/python/topi/cuda/reduction.py
+2
-2
topi/python/topi/hls/nn.py
+5
-5
topi/python/topi/intel_graphics/depthwise_conv2d.py
+1
-1
topi/python/topi/opengl/conv2d_nchw.py
+1
-1
topi/python/topi/opengl/dense.py
+1
-1
topi/python/topi/opengl/pooling.py
+2
-2
topi/python/topi/x86/binary_dense.py
+1
-1
topi/python/topi/x86/bitserial_dense.py
+1
-1
topi/python/topi/x86/pooling.py
+2
-2
topi/python/topi/x86/reduction.py
+2
-2
No files found.
python/tvm/intrin.py
View file @
7cf1eadc
...
@@ -575,7 +575,7 @@ def register_intrin_rule(target, intrin, f=None, override=False):
...
@@ -575,7 +575,7 @@ def register_intrin_rule(target, intrin, f=None, override=False):
The name of codegen target.
The name of codegen target.
intrin : str
intrin : str
The name of the in
s
trinsic.
The name of the intrinsic.
f : function, optional
f : function, optional
The function to be registered.
The function to be registered.
...
...
src/api/api_pass.cc
View file @
7cf1eadc
...
@@ -18,7 +18,7 @@
...
@@ -18,7 +18,7 @@
*/
*/
/*!
/*!
* Exposre of pass functions.
* Expos
u
re of pass functions.
* \file api_pass.cc
* \file api_pass.cc
*/
*/
#include <tvm/expr.h>
#include <tvm/expr.h>
...
...
src/pass/lower_warp_memory.cc
View file @
7cf1eadc
...
@@ -46,9 +46,9 @@ namespace ir {
...
@@ -46,9 +46,9 @@ namespace ir {
// This requires us to do the following rewriting:
// This requires us to do the following rewriting:
// - Rewrite allocation to use local memory.
// - Rewrite allocation to use local memory.
// - Rewrite store of warp memory to local store.
// - Rewrite store of warp memory to local store.
// - Rewrite load of war
o
memory to local plus a shuffle.
// - Rewrite load of war
p
memory to local plus a shuffle.
//
//
// Define a generic shuffle in
s
trinsic warp_shuffle(data, warp_index).
// Define a generic shuffle intrinsic warp_shuffle(data, warp_index).
// We can use the following rewriting rule
// We can use the following rewriting rule
//
//
// Before rewrite,
// Before rewrite,
...
...
topi/python/topi/arm_cpu/bitserial_dense.py
View file @
7cf1eadc
...
@@ -158,7 +158,7 @@ def schedule_bitserial_dense(cfg, outs):
...
@@ -158,7 +158,7 @@ def schedule_bitserial_dense(cfg, outs):
return
s
return
s
def
traverse
(
op
):
def
traverse
(
op
):
"""Internal trav
s
erse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
# inline all one-to-one-mapping operators except the last stage (output)
if
tag
.
is_broadcast
(
op
.
tag
)
or
'elemwise'
in
op
.
tag
:
if
tag
.
is_broadcast
(
op
.
tag
)
or
'elemwise'
in
op
.
tag
:
if
op
not
in
s
.
outputs
:
if
op
not
in
s
.
outputs
:
...
...
topi/python/topi/bifrost/depthwise_conv2d.py
View file @
7cf1eadc
...
@@ -104,7 +104,7 @@ def schedule_depthwise_conv2d_nchw(outs):
...
@@ -104,7 +104,7 @@ def schedule_depthwise_conv2d_nchw(outs):
s
[
conv
]
.
compute_at
(
s
[
output
],
ji
)
s
[
conv
]
.
compute_at
(
s
[
output
],
ji
)
def
traverse
(
op
):
def
traverse
(
op
):
"""Internal trav
s
erse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
# inline all one-to-one-mapping operators except the last stage (output)
if
tag
.
is_broadcast
(
op
.
tag
):
if
tag
.
is_broadcast
(
op
.
tag
):
if
op
not
in
s
.
outputs
:
if
op
not
in
s
.
outputs
:
...
...
topi/python/topi/cuda/dense.py
View file @
7cf1eadc
...
@@ -105,7 +105,7 @@ def schedule_dense(cfg, outs):
...
@@ -105,7 +105,7 @@ def schedule_dense(cfg, outs):
scheduled_ops
=
[]
scheduled_ops
=
[]
def
traverse
(
OP
):
def
traverse
(
OP
):
"""Internal trav
s
erse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
# inline all one-to-one-mapping operators except the last stage (output)
if
tag
.
is_broadcast
(
OP
.
tag
):
if
tag
.
is_broadcast
(
OP
.
tag
):
if
OP
not
in
s
.
outputs
:
if
OP
not
in
s
.
outputs
:
...
...
topi/python/topi/cuda/depthwise_conv2d.py
View file @
7cf1eadc
...
@@ -192,7 +192,7 @@ def schedule_depthwise_conv2d_nhwc(outs):
...
@@ -192,7 +192,7 @@ def schedule_depthwise_conv2d_nhwc(outs):
scheduled_ops
=
[]
scheduled_ops
=
[]
def
traverse
(
OP
):
def
traverse
(
OP
):
"""Internal trav
s
erse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
# inline all one-to-one-mapping operators except the last stage (output)
if
tag
.
is_broadcast
(
OP
.
tag
):
if
tag
.
is_broadcast
(
OP
.
tag
):
if
OP
not
in
s
.
outputs
:
if
OP
not
in
s
.
outputs
:
...
...
topi/python/topi/cuda/pooling.py
View file @
7cf1eadc
...
@@ -68,7 +68,7 @@ def schedule_adaptive_pool(outs):
...
@@ -68,7 +68,7 @@ def schedule_adaptive_pool(outs):
scheduled_ops
=
[]
scheduled_ops
=
[]
def
traverse
(
OP
):
def
traverse
(
OP
):
"""Internal trav
s
erse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
# inline all one-to-one-mapping operators except the last stage (output)
if
tag
.
is_broadcast
(
OP
.
tag
):
if
tag
.
is_broadcast
(
OP
.
tag
):
if
OP
not
in
s
.
outputs
:
if
OP
not
in
s
.
outputs
:
...
@@ -131,7 +131,7 @@ def schedule_pool(outs, layout):
...
@@ -131,7 +131,7 @@ def schedule_pool(outs, layout):
scheduled_ops
=
[]
scheduled_ops
=
[]
def
traverse
(
OP
):
def
traverse
(
OP
):
"""Internal trav
s
erse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
# inline all one-to-one-mapping operators except the last stage (output)
if
tag
.
is_broadcast
(
OP
.
tag
):
if
tag
.
is_broadcast
(
OP
.
tag
):
if
OP
not
in
s
.
outputs
:
if
OP
not
in
s
.
outputs
:
...
...
topi/python/topi/cuda/reduction.py
View file @
7cf1eadc
...
@@ -109,7 +109,7 @@ def schedule_reduce(outs):
...
@@ -109,7 +109,7 @@ def schedule_reduce(outs):
scheduled_ops
=
[]
scheduled_ops
=
[]
def
traverse_before_reduce
(
operator
):
def
traverse_before_reduce
(
operator
):
"""Internal trav
s
erse function"""
"""Internal traverse function"""
if
isinstance
(
operator
,
tvm
.
tensor
.
PlaceholderOp
):
if
isinstance
(
operator
,
tvm
.
tensor
.
PlaceholderOp
):
return
return
if
tag
.
is_injective
(
operator
.
tag
):
if
tag
.
is_injective
(
operator
.
tag
):
...
@@ -123,7 +123,7 @@ def schedule_reduce(outs):
...
@@ -123,7 +123,7 @@ def schedule_reduce(outs):
scheduled_ops
.
append
(
operator
)
scheduled_ops
.
append
(
operator
)
def
traverse_after_reduce
(
operator
):
def
traverse_after_reduce
(
operator
):
"""Internal trav
s
erse function"""
"""Internal traverse function"""
if
tag
.
is_broadcast
(
operator
.
tag
):
if
tag
.
is_broadcast
(
operator
.
tag
):
if
operator
not
in
scheduled_ops
:
if
operator
not
in
scheduled_ops
:
schedule_injective_from_existing
(
sch
,
operator
.
output
(
0
))
schedule_injective_from_existing
(
sch
,
operator
.
output
(
0
))
...
...
topi/python/topi/hls/nn.py
View file @
7cf1eadc
...
@@ -28,7 +28,7 @@ def _schedule_conv2d(outs):
...
@@ -28,7 +28,7 @@ def _schedule_conv2d(outs):
tvm
.
schedule
.
AutoInlineInjective
(
s
)
tvm
.
schedule
.
AutoInlineInjective
(
s
)
def
traverse
(
OP
):
def
traverse
(
OP
):
"""Internal trav
s
erse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
# inline all one-to-one-mapping operators except the last stage (output)
if
tag
.
is_injective
(
OP
.
tag
):
if
tag
.
is_injective
(
OP
.
tag
):
if
OP
not
in
s
.
outputs
:
if
OP
not
in
s
.
outputs
:
...
@@ -214,7 +214,7 @@ def schedule_reduce(outs):
...
@@ -214,7 +214,7 @@ def schedule_reduce(outs):
tvm
.
schedule
.
AutoInlineInjective
(
s
)
tvm
.
schedule
.
AutoInlineInjective
(
s
)
def
traverse
(
OP
):
def
traverse
(
OP
):
"""Internal trav
s
erse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
# inline all one-to-one-mapping operators except the last stage (output)
if
tag
.
is_broadcast
(
OP
.
tag
):
if
tag
.
is_broadcast
(
OP
.
tag
):
if
OP
not
in
s
.
outputs
:
if
OP
not
in
s
.
outputs
:
...
@@ -306,7 +306,7 @@ def schedule_dense(outs):
...
@@ -306,7 +306,7 @@ def schedule_dense(outs):
tvm
.
schedule
.
AutoInlineInjective
(
s
)
tvm
.
schedule
.
AutoInlineInjective
(
s
)
def
traverse
(
OP
):
def
traverse
(
OP
):
"""Internal trav
s
erse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
# inline all one-to-one-mapping operators except the last stage (output)
if
tag
.
is_broadcast
(
OP
.
tag
):
if
tag
.
is_broadcast
(
OP
.
tag
):
if
OP
not
in
s
.
outputs
:
if
OP
not
in
s
.
outputs
:
...
@@ -350,7 +350,7 @@ def schedule_pool(outs, layout):
...
@@ -350,7 +350,7 @@ def schedule_pool(outs, layout):
tvm
.
schedule
.
AutoInlineInjective
(
s
)
tvm
.
schedule
.
AutoInlineInjective
(
s
)
def
traverse
(
OP
):
def
traverse
(
OP
):
"""Internal trav
s
erse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
# inline all one-to-one-mapping operators except the last stage (output)
if
tag
.
is_broadcast
(
OP
.
tag
):
if
tag
.
is_broadcast
(
OP
.
tag
):
if
OP
not
in
s
.
outputs
:
if
OP
not
in
s
.
outputs
:
...
@@ -394,7 +394,7 @@ def schedule_adaptive_pool(outs):
...
@@ -394,7 +394,7 @@ def schedule_adaptive_pool(outs):
tvm
.
schedule
.
AutoInlineInjective
(
s
)
tvm
.
schedule
.
AutoInlineInjective
(
s
)
def
traverse
(
OP
):
def
traverse
(
OP
):
"""Internal trav
s
erse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
# inline all one-to-one-mapping operators except the last stage (output)
if
tag
.
is_broadcast
(
OP
.
tag
):
if
tag
.
is_broadcast
(
OP
.
tag
):
if
OP
not
in
s
.
outputs
:
if
OP
not
in
s
.
outputs
:
...
...
topi/python/topi/intel_graphics/depthwise_conv2d.py
View file @
7cf1eadc
...
@@ -193,7 +193,7 @@ def schedule_depthwise_conv2d_nhwc(outs):
...
@@ -193,7 +193,7 @@ def schedule_depthwise_conv2d_nhwc(outs):
scheduled_ops
=
[]
scheduled_ops
=
[]
def
traverse
(
OP
):
def
traverse
(
OP
):
"""Internal trav
s
erse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
# inline all one-to-one-mapping operators except the last stage (output)
if
tag
.
is_broadcast
(
OP
.
tag
):
if
tag
.
is_broadcast
(
OP
.
tag
):
if
OP
not
in
s
.
outputs
:
if
OP
not
in
s
.
outputs
:
...
...
topi/python/topi/opengl/conv2d_nchw.py
View file @
7cf1eadc
...
@@ -49,7 +49,7 @@ def schedule_conv2d_nchw(outs):
...
@@ -49,7 +49,7 @@ def schedule_conv2d_nchw(outs):
s
[
data
]
.
opengl
()
s
[
data
]
.
opengl
()
def
traverse
(
OP
):
def
traverse
(
OP
):
"""Internal trav
s
erse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
# inline all one-to-one-mapping operators except the last stage (output)
if
tag
.
is_broadcast
(
OP
.
tag
):
if
tag
.
is_broadcast
(
OP
.
tag
):
if
OP
not
in
s
.
outputs
:
if
OP
not
in
s
.
outputs
:
...
...
topi/python/topi/opengl/dense.py
View file @
7cf1eadc
...
@@ -49,7 +49,7 @@ def schedule_dense(outs):
...
@@ -49,7 +49,7 @@ def schedule_dense(outs):
s
[
Out
]
.
opengl
()
s
[
Out
]
.
opengl
()
def
traverse
(
OP
):
def
traverse
(
OP
):
"""Internal trav
s
erse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
# inline all one-to-one-mapping operators except the last stage (output)
if
tag
.
is_broadcast
(
OP
.
tag
):
if
tag
.
is_broadcast
(
OP
.
tag
):
if
OP
not
in
s
.
outputs
:
if
OP
not
in
s
.
outputs
:
...
...
topi/python/topi/opengl/pooling.py
View file @
7cf1eadc
...
@@ -48,7 +48,7 @@ def schedule_adaptive_pool(outs):
...
@@ -48,7 +48,7 @@ def schedule_adaptive_pool(outs):
s
[
Out
]
.
opengl
()
s
[
Out
]
.
opengl
()
def
traverse
(
OP
):
def
traverse
(
OP
):
"""Internal trav
s
erse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
# inline all one-to-one-mapping operators except the last stage (output)
if
tag
.
is_broadcast
(
OP
.
tag
):
if
tag
.
is_broadcast
(
OP
.
tag
):
if
OP
not
in
s
.
outputs
:
if
OP
not
in
s
.
outputs
:
...
@@ -102,7 +102,7 @@ def schedule_pool(outs, layout):
...
@@ -102,7 +102,7 @@ def schedule_pool(outs, layout):
s
[
Out
]
.
opengl
()
s
[
Out
]
.
opengl
()
def
traverse
(
OP
):
def
traverse
(
OP
):
"""Internal trav
s
erse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
# inline all one-to-one-mapping operators except the last stage (output)
if
tag
.
is_broadcast
(
OP
.
tag
):
if
tag
.
is_broadcast
(
OP
.
tag
):
if
OP
not
in
s
.
outputs
:
if
OP
not
in
s
.
outputs
:
...
...
topi/python/topi/x86/binary_dense.py
View file @
7cf1eadc
...
@@ -52,7 +52,7 @@ def schedule_binary_dense(outs):
...
@@ -52,7 +52,7 @@ def schedule_binary_dense(outs):
s
[
Out
]
.
vectorize
(
xi
)
s
[
Out
]
.
vectorize
(
xi
)
def
traverse
(
OP
):
def
traverse
(
OP
):
"""Internal trav
s
erse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
# inline all one-to-one-mapping operators except the last stage (output)
if
tag
.
is_broadcast
(
OP
.
tag
):
if
tag
.
is_broadcast
(
OP
.
tag
):
if
OP
not
in
s
.
outputs
:
if
OP
not
in
s
.
outputs
:
...
...
topi/python/topi/x86/bitserial_dense.py
View file @
7cf1eadc
...
@@ -69,7 +69,7 @@ def schedule_bitserial_dense(cfg, outs):
...
@@ -69,7 +69,7 @@ def schedule_bitserial_dense(cfg, outs):
return
s
return
s
def
traverse
(
op
):
def
traverse
(
op
):
"""Internal trav
s
erse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
# inline all one-to-one-mapping operators except the last stage (output)
if
tag
.
is_broadcast
(
op
.
tag
)
or
'elemwise'
in
op
.
tag
:
if
tag
.
is_broadcast
(
op
.
tag
)
or
'elemwise'
in
op
.
tag
:
if
op
not
in
s
.
outputs
:
if
op
not
in
s
.
outputs
:
...
...
topi/python/topi/x86/pooling.py
View file @
7cf1eadc
...
@@ -88,7 +88,7 @@ def schedule_pool(outs, layout):
...
@@ -88,7 +88,7 @@ def schedule_pool(outs, layout):
_parallel_sch
(
s
[
Pool
],
outs
[
0
]
.
shape
,
do_vectorize
)
_parallel_sch
(
s
[
Pool
],
outs
[
0
]
.
shape
,
do_vectorize
)
def
traverse
(
OP
):
def
traverse
(
OP
):
"""Internal trav
s
erse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
# inline all one-to-one-mapping operators except the last stage (output)
if
tag
.
is_broadcast
(
OP
.
tag
):
if
tag
.
is_broadcast
(
OP
.
tag
):
if
OP
not
in
s
.
outputs
:
if
OP
not
in
s
.
outputs
:
...
@@ -137,7 +137,7 @@ def schedule_adaptive_pool(outs):
...
@@ -137,7 +137,7 @@ def schedule_adaptive_pool(outs):
scheduled_ops
=
[]
scheduled_ops
=
[]
def
traverse
(
OP
):
def
traverse
(
OP
):
"""Internal trav
s
erse function"""
"""Internal traverse function"""
# inline all one-to-one-mapping operators except the last stage (output)
# inline all one-to-one-mapping operators except the last stage (output)
if
tag
.
is_broadcast
(
OP
.
tag
):
if
tag
.
is_broadcast
(
OP
.
tag
):
if
OP
not
in
s
.
outputs
:
if
OP
not
in
s
.
outputs
:
...
...
topi/python/topi/x86/reduction.py
View file @
7cf1eadc
...
@@ -78,7 +78,7 @@ def schedule_reduce(outs):
...
@@ -78,7 +78,7 @@ def schedule_reduce(outs):
scheduled_ops
=
[]
scheduled_ops
=
[]
def
traverse_before_reduce
(
operator
):
def
traverse_before_reduce
(
operator
):
"""Internal trav
s
erse function"""
"""Internal traverse function"""
if
isinstance
(
operator
,
tvm
.
tensor
.
PlaceholderOp
):
if
isinstance
(
operator
,
tvm
.
tensor
.
PlaceholderOp
):
return
return
if
tag
.
is_injective
(
operator
.
tag
):
if
tag
.
is_injective
(
operator
.
tag
):
...
@@ -92,7 +92,7 @@ def schedule_reduce(outs):
...
@@ -92,7 +92,7 @@ def schedule_reduce(outs):
scheduled_ops
.
append
(
operator
)
scheduled_ops
.
append
(
operator
)
def
traverse_after_reduce
(
operator
):
def
traverse_after_reduce
(
operator
):
"""Internal trav
s
erse function"""
"""Internal traverse function"""
if
tag
.
is_broadcast
(
operator
.
tag
):
if
tag
.
is_broadcast
(
operator
.
tag
):
if
operator
not
in
scheduled_ops
:
if
operator
not
in
scheduled_ops
:
generic
.
schedule_injective_from_existing
(
sch
,
operator
)
generic
.
schedule_injective_from_existing
(
sch
,
operator
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment