Fix lint

apache · Feb 23, 2021 · 7173b3d · 7173b3d
1 parent 1602d9e
commit 7173b3d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/topi/cuda/unique.py b/python/tvm/topi/cuda/unique.py
@@ -54,7 +54,7 @@ def _calc_adjacent_diff_ir(data, adjacent_diff):
 def _calc_num_unique(data):
     output = output_tensor((1,), "int32")
     for i in bind("threadIdx.x", 1):
-        output[0] = data[data.shape[0] - 1] + int32(1)
+        output[i] = data[data.shape[0] - 1] + int32(1)
     return output