triton.language.device_assert

triton.language.device_assert(cond, msg='')

在设备运行时断言条件。需要将环境变量TRITON_DEBUG设置为除0以外的值才能生效。

使用Python的assert语句与调用此函数效果相同，区别在于第二个参数必须提供且必须是一个字符串，例如assert pid == 0, "pid != 0"。必须设置环境变量才能使这个assert语句生效。

tl.device_assert(pid == 0)
assert pid == 0, f"pid != 0"

Parameters:

• cond - 要断言的条件。这必须是一个布尔张量。

• msg – 如果断言失败时要打印的消息。这必须是一个字符串字面量。