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"

参数:

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

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