triton.language.device_assert

triton.language.device_assert(cond, msg='', _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 – 要断言的条件。这必须是布尔张量。

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