simplify bin search

apache · Feb 9, 2021 · 3a8eef6 · 3a8eef6
1 parent 38b1539
commit 3a8eef6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/unittest/test_tir_ir_builder.py b/tests/python/unittest/test_tir_ir_builder.py
@@ -419,7 +419,7 @@ def binary_search(ib, n, i, Aptr, Bptr, Cptr):
  v = Bptr[i]
 
  with ib.while_loop(lo[0] < hi[0]):
- mid = lo[0] + tvm.tir.floordiv(hi[0] - lo[0], 2).astype("int32")
+ mid = lo[0] + (hi[0] - lo[0] >> 1)
  with ib.if_scope(Aptr[mid] < v):
  lo[0] = mid + 1
  with ib.else_scope():