triton.language.device_assert

triton.language.device_assert(cond, msg='', mask=None, _semantic=None)

在运行时从设备端断言条件。需要将环境变量 TRITON_DEBUG 设置为非 0 的值，此函数才会生效。

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

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

参数:

• cond – 要断言的条件。此参数必须是布尔张量（boolean tensor）。

• msg – 如果断言失败要打印的消息。此参数必须是字符串字面量（string literal）。