fix lint

apache · Nov 3, 2020 · 71c032a · 71c032a
1 parent fadbfe1
commit 71c032a
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 8 deletions.
diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py
@@ -2235,13 +2235,9 @@ def conditionally_squeeze_scalar(x):
         one = _op.const(np.array([1]), dtype="int64")
         two = _op.const(np.array([2]), dtype="int64")
         three = _op.const(np.array([3]), dtype="int64")
-        two_ones = _op.const(np.array([1, 1]), dtype="int64")
         three_ones = _op.const(np.array([1, 1, 1]), dtype="int64")
         four_ones = _op.const(np.array([1, 1, 1, 1]), dtype="int64")
 
-        def pad_last_dim(x):
-            return _op.expand_dims(x, -1, 1)
-
         # First Loop Vars
         i = _expr.var("i", shape=(1,), dtype="int64")
         scores_var = _expr.var("scores_var", shape=(_ty.Any(), _ty.Any(), _ty.Any()), dtype=dtype)

diff --git a/python/tvm/topi/cuda/nms.py b/python/tvm/topi/cuda/nms.py
@@ -98,10 +98,6 @@ def get_valid_counts_ir(
     valid_count = ib.buffer_ptr(valid_count)
     out = ib.buffer_ptr(out)
     out_indices = ib.buffer_ptr(out_indices)
-    atomic_add_return = ib.allocate(
-        valid_count.dtype, (1,), name="atomic_add_return", scope="local"
-    )
-    one_count = tvm.tir.const(1, dtype=valid_count.dtype)
     one = tvm.tir.const(1, dtype=out.dtype)
     if isinstance(score_threshold, float):
         score_threshold = tvm.ir.make_node("FloatImm", dtype="float32", value=score_threshold)