TRESHAPE

指令示意图

TRESHAPE tile operation

简介

将 Tile 重新解释为另一种 Tile 类型/形状，同时保留底层字节。

数学语义

除非另有说明, semantics are defined over the valid region and target-dependent behavior is marked as implementation-defined.

汇编语法

PTO-AS 形式：参见 PTO-AS Specification.

%dst = treshape %src : !pto.tile<...>

AS Level 1 (SSA)

%dst = pto.treshape %src : !pto.tile<...> -> !pto.tile<...>

AS Level 2 (DPS)

pto.treshape ins(%src : !pto.tile_buf<...>) outs(%dst : !pto.tile_buf<...>)

AS Level 1（SSA）

%dst = pto.treshape %src : !pto.tile<...> -> !pto.tile<...>

AS Level 2（DPS）

pto.treshape ins(%src : !pto.tile_buf<...>) outs(%dst : !pto.tile_buf<...>)

C++ 内建接口

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

template <typename TileDataOut, typename TileDataIn, typename... WaitEvents>
PTO_INST RecordEvent TRESHAPE(TileDataOut &dst, TileDataIn &src, WaitEvents &... events);

约束

Enforced by TRESHAPE_IMPL:

  • Tile type must match: TileDataIn::Loc == TileDataOut::Loc.
  • Total byte size must match: sizeof(InElem) * InNumel == sizeof(OutElem) * OutNumel.
  • No boxed/non-boxed conversion:
    • cannot reshape between SLayout::NoneBox and boxed layouts.

示例

#include <pto/pto-inst.hpp>

using namespace pto;

void example() {
  using Src = Tile<TileType::Vec, float, 16, 16>;
  using Dst = Tile<TileType::Vec, float, 8, 32>;
  static_assert(Src::Numel == Dst::Numel);

  Src src;
  Dst dst;
  TRESHAPE(dst, src);
}