Skip to content

Commit

Permalink
fix lint
Browse files Browse the repository at this point in the history
  • Loading branch information
eric-haibin-lin authored Jul 4, 2020
1 parent 4912bfe commit b34f9f9
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions python/tvm/relay/frontend/mxnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -913,9 +913,9 @@ def _mx_amp_multicast(inputs, attrs):
has_float32 = any(x == "float32" for x in dtypes)
dtype = dtypes[0]
if cast_narrow and has_float16:
dtype = 'float16'
dtype = 'float16'
if not cast_narrow and has_float32:
dtype = 'float32'
dtype = 'float32'
return [_op.cast(x, dtype) for x in inputs]

def _mx_grid_generator(inputs, attrs):
Expand Down

0 comments on commit b34f9f9

Please sign in to comment.