triton.language.device_print

1. 函数概述

device_print 用于在NPU运行时从设备端打印信息，与static_print不同，这是在内核执行时实时输出信息。 第一个参数必须为string, 后面的参数必须为scalars或者tensors。 使用device_print需要将环境变量TRITON_DEVICE_PRINT的值设置为True。

triton.language.device_print(prefix, *args, hex=False, _semantic=None)

2. 规格

2.1 参数说明

参数	类型	默认值	含义说明
prefix	str	必需	打印值之前的前缀字符串
args	tensor/scalar	必需	要打印的值，可以是任意张量或标量
hex	bool	False	是否以十六进制格式打印所有值
_semantic	-	-	保留参数，暂不支持外部调用

2.2.1 Data Type支持

A3：

	int8	int16	int32	uint8	uint16	uint32	uint64	int64	fp16	fp32	fp64	bf16	bool
GPU	✓	✓	✓	✓	✓	✓	✓	✓	✓	✓	✓	✓	✓
Ascend A2/A3	✓	✓	✓	×	×	×	×	✓	✓	✓	×	✓	✓

2.2.2 Shape 支持

	支持维度范围
GPU	仅支持 1~5维 tensor
Ascend	仅支持 1~5维 tensor

2.3 特殊限制说明

相对社区能力缺失且无法实现

Ascend 对比 GPU 缺失uint8、uint16、uint32、uint64、fp64的支持能力（硬件限制）。

2.4 使用方法

注意：prefix字符串前缀在使用device_print时是必须加上的，否则会导致编译错误。

import triton
import triton.language as tl

@triton.jit
def kernel(x_ptr):
    idx = tl.arange(0,3)
    idy = tl.arange(0,4)
    offset = idx[:,None] * 4 + idy[None,:]
    val = tl.load(x_ptr + offset)
    # 打印二维张量val的值
    tl.device_print("val:",val)