MCPcopy Create free account
hub / github.com/apache/tvm / compare

Function compare

python/tvm/topi/gpu/sort.py:214–220  ·  view source on GitHub ↗

Compare a and b in proper ascending or descending order

(a, b)

Source from the content-addressed store, hash-verified

212 nthread_bx = ceil_div(size, nthread_tx)
213
214 def compare(a, b):
215 """Compare a and b in proper ascending or descending order"""
216 if is_ascend:
217 out = a <= b
218 else:
219 out = b <= a
220 return out
221
222 # Sort the lower levels of the merge using odd-even sort, it's fast for small inputs
223 lower_lim = ceil_log2(block_size)

Callers 2

get_merge_beginFunction · 0.70
serial_mergeFunction · 0.70

Calls

no outgoing calls

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…