TPRINT¶

指令示意图¶

简介¶

调试/打印 Tile 中的元素（实现定义）。

从设备代码直接打印 Tile 或 GlobalTensor 的内容以用于调试目的。

TPRINT 指令输出存储在 Tile 或 GlobalTensor 中的数据的逻辑视图。它支持常见的数据类型（例如 float、half、int8、uint32）和多种内存布局（GlobalTensor 的 ND、DN、NZ；片上缓冲区的向量 tiles）。

重要: - 此指令仅用于开发和调试。 - 它会产生显著的运行时开销，不得在生产 kernel 中使用。 - 如果输出超过内部打印缓冲区，可能会被截断。 - 需要 CCE 编译选项 -D_DEBUG --cce-enable-print（参见 行为）。

数学语义¶

除非另有说明，语义在有效区域上定义，目标相关的行为标记为实现定义。

汇编语法¶

PTO-AS 形式：参见 PTO-AS 规范。

tprint %src : !pto.tile<...> | !pto.global<...>

AS Level 1（SSA）¶

pto.tprint %src : !pto.tile<...> | !pto.partition_tensor_view<MxNxdtype> -> ()

AS Level 2（DPS）¶

pto.tprint ins(%src : !pto.tile_buf<...> | !pto.partition_tensor_view<MxNxdtype>)

C++ 内建接口¶

声明于 include/pto/common/pto_instr.hpp：

template <typename TileData, typename... WaitEvents>
PTO_INST RecordEvent TPRINT(TileData &src, WaitEvents &... events);

支持的 T 类型¶

• Tile：必须是向量 tile（TileType::Vec），具有支持的元素类型。
• GlobalTensor：必须使用布局 ND、DN 或 NZ，并具有支持的元素类型。

约束¶

• 支持的元素类型:
  • 浮点数：float、half
  • 有符号整数：int8_t、int16_t、int32_t
  • 无符号整数：uint8_t、uint16_t、uint32_t
• 对于 Tiles：TileData::Loc == TileType::Vec（仅向量 tiles 可打印）。
• 对于 GlobalTensor：布局必须是 Layout::ND、Layout::DN 或 Layout::NZ 之一。

示例¶

Print a Tile¶

#include <pto/pto-inst.hpp>

PTO_INTERNAL void DebugTile(__gm__ float *src) {
  using ValidSrcShape = TileShape2D<float, 16, 16>;
  using NDSrcShape = BaseShape2D<float, 32, 32>;
  using GlobalDataSrc = GlobalTensor<float, ValidSrcShape, NDSrcShape>;
  GlobalDataSrc srcGlobal(src);

  using srcTileData = Tile<TileType::Vec, float, 16, 16>;
  srcTileData srcTile;
  TASSIGN(srcTile, 0x0);

  TLOAD(srcTile, srcGlobal);
  TPRINT(srcTile);
}

Print a GlobalTensor¶

#include <pto/pto-inst.hpp>

PTO_INTERNAL void DebugGlobalTensor(__gm__ float *src) {
  using ValidSrcShape = TileShape2D<float, 16, 16>;
  using NDSrcShape = BaseShape2D<float, 32, 32>;
  using GlobalDataSrc = GlobalTensor<float, ValidSrcShape, NDSrcShape>;
  GlobalDataSrc srcGlobal(src);

  TPRINT(srcGlobal);
}

汇编示例（ASM）¶

自动模式¶

# 自动模式：由编译器/运行时负责资源放置与调度。
pto.tprint %src : !pto.tile<...> | !pto.partition_tensor_view<MxNxdtype> -> ()

手动模式¶

# 手动模式：先显式绑定资源，再发射指令。
# 可选（当该指令包含 tile 操作数时）：
# pto.tassign %arg0, @tile(0x1000)
# pto.tassign %arg1, @tile(0x2000)
pto.tprint %src : !pto.tile<...> | !pto.partition_tensor_view<MxNxdtype> -> ()

PTO 汇编形式¶

pto.tprint %src : !pto.tile<...> | !pto.partition_tensor_view<MxNxdtype> -> ()
# AS Level 2 (DPS)
pto.tprint ins(%src : !pto.tile_buf<...> | !pto.partition_tensor_view<MxNxdtype>)