Skip to content

Commit

Permalink
[lint] Fix black whitespace errors
Browse files Browse the repository at this point in the history
Change-Id: I927b43df95a8db8b042bc3cf2a1f23739d102b9d
  • Loading branch information
ekalda committed May 26, 2021
1 parent 69e56c6 commit 7e630e3
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions python/tvm/topi/mali/conv2d.py
Expand Up @@ -607,7 +607,7 @@ def conv2d_winograd_nhwc_mali(

##### SCHECULE UTILITIES #####
def tile_and_bind(s, tensor, y, x, y_factor, x_factor=None):
""" tile and bind to GPU threads """
"""tile and bind to GPU threads"""
x_factor = x_factor or y_factor
yo, xo, yi, xi = s[tensor].tile(y, x, y_factor, x_factor)
s[tensor].bind(xo, te.thread_axis("blockIdx.x"))
Expand All @@ -618,7 +618,7 @@ def tile_and_bind(s, tensor, y, x, y_factor, x_factor=None):


def tile_and_bind3d(s, tensor, z, y, x, z_factor=2, y_factor=None, x_factor=None):
""" tile and bind 3d """
"""tile and bind 3d"""
y_factor = y_factor or z_factor
x_factor = x_factor or y_factor
zo, zi = s[tensor].split(z, z_factor)
Expand Down

0 comments on commit 7e630e3

Please sign in to comment.