diff --git a/python/tvm/topi/cuda/scatter_elements.py b/python/tvm/topi/cuda/scatter_elements.py index 3e6eebf81915..8ed3e2972081 100644 --- a/python/tvm/topi/cuda/scatter_elements.py +++ b/python/tvm/topi/cuda/scatter_elements.py @@ -102,7 +102,6 @@ def gen_ir(data, indices, updates, out, axis): elif i > axis: ind_after_axis_range *= value ind_before_axis_stride = ind_axis_range * ind_after_axis_range - ind_full_range = ind_before_axis_range * ind_before_axis_stride ind_full_range_excl_axis = ind_before_axis_range * ind_after_axis_range max_threads = int(tvm.target.Target.current(allow_none=False).max_num_threads)