tvm_tirx.tirx.analysis

Contents

tvm_tirx.tirx.analysis#

Analyze tensor-level IR, including memory access, control flow, and well-formedness.

Functions#

verify_memory(func)

Verify if func contains illegal host side direct memory access.

undefined_vars(node[, defs])

Find undefined vars in a TIR statement or expression.

check_well_formed(obj[, assert_mode])

Check whether a TIRx function is well formed.

detect_linear_equation(expr, var_list)

Match expr = sum_{i=0}^{n-1} var[i] * coeff[i] + coeff[n]

detect_clip_bound(expr, var_list)

Detect if expression corresponds to clip bound of the vars
Contents