tvm_tirx.tirx.script.select

Contents

tvm_tirx.tirx.script.select#

tvm_tirx.tirx.script.select(condition, true_value, false_value)#

Construct a scalar conditional or select a host value.

Parameters:

  • condition (Expr or Python value) – Scalar IR condition or host truth value. Conditions implementing __tvm_ffi_object__ are normalized before selection.

  • true_value (Expr or Python value) – Value selected when the condition is true.

  • false_value (Expr or Python value) – Value selected when the condition is false.

Returns:

result (Expr or Python value) – A primitive if_then_else expression, or the selected host object.

Notes

Both Python value arguments are constructed before this call. The generated IR conditional evaluates only its selected arm at runtime.

Contents