tilelang.cuda.debug¶

Functions¶

get_stack_str(msg[, stacklevel])

device_assert(condition[, msg, no_stack_info])

Device-side assert emulation for CUDA targets.

Module Contents¶

tilelang.cuda.debug.get_stack_str(msg, stacklevel=1)¶
tilelang.cuda.debug.device_assert(condition, msg='', no_stack_info=False)¶

Device-side assert emulation for CUDA targets.

Parameters:

  • condition (tvm.tirx.PrimExpr)

  • msg (str)