TMINS¶
Tile Operation Diagram¶
Introduction¶
Elementwise minimum of a tile and a scalar.
Math Interpretation¶
For each element (i, j) in the valid region:
\[ \mathrm{dst}_{i,j} = \min(\mathrm{src}_{i,j}, \mathrm{scalar}) \]
Assembly Syntax¶
PTO-AS form: see PTO-AS Specification.
Synchronous form:
%dst = tmins %src, %scalar : !pto.tile<...>, f32
IR Level 1 (SSA)¶
%dst = pto.tmins %src, %scalar : (!pto.tile<...>, dtype) -> !pto.tile<...>
IR Level 2 (DPS)¶
pto.tmins ins(%src, %scalar : !pto.tile_buf<...>, dtype) outs(%dst : !pto.tile_buf<...>)
C++ Intrinsic¶
Declared in include/pto/common/pto_instr.hpp:
template <typename TileData, typename T, typename... WaitEvents>
PTO_INST RecordEvent TMINS(TileData& dst, TileData& src, T scalar, WaitEvents&... events);
Constraints¶
- Implementation checks (A2A3):
- No additional
static_assert/PTO_ASSERTchecks are enforced byTMINS_IMPLbeyond the generic Tile type invariants. - Implementation checks (A5):
TileData::DTypemust be one of:uint8_t,int8_t,uint16_t,int16_t,uint32_t,int32_t,half,float,bfloat16_t.- Tile location must be vector (
TileData::Loc == TileType::Vec). - Static valid bounds:
TileData::ValidRow <= TileData::RowsandTileData::ValidCol <= TileData::Cols. - Runtime:
src.GetValidCol() == dst.GetValidCol(). - Valid region:
- The op uses
dst.GetValidRow()/dst.GetValidCol()as the iteration domain.
Examples¶
Auto¶
#include <pto/pto-inst.hpp>
using namespace pto;
void example_auto() {
using TileT = Tile<TileType::Vec, float, 16, 16>;
TileT src, dst;
TMINS(dst, src, 0.0f);
}
Manual¶
#include <pto/pto-inst.hpp>
using namespace pto;
void example_manual() {
using TileT = Tile<TileType::Vec, float, 16, 16>;
TileT src, dst;
TASSIGN(src, 0x1000);
TASSIGN(dst, 0x2000);
TMINS(dst, src, 0.0f);
}
ASM Form Examples¶
Auto Mode¶
# Auto mode: compiler/runtime-managed placement and scheduling.
%dst = pto.tmins %src, %scalar : (!pto.tile<...>, dtype) -> !pto.tile<...>
Manual Mode¶
# Manual mode: bind resources explicitly before issuing the instruction.
# Optional for tile operands:
# pto.tassign %arg0, @tile(0x1000)
# pto.tassign %arg1, @tile(0x2000)
%dst = pto.tmins %src, %scalar : (!pto.tile<...>, dtype) -> !pto.tile<...>
PTO Assembly Form¶
%dst = tmins %src, %scalar : !pto.tile<...>, f32
# IR Level 2 (DPS)
pto.tmins ins(%src, %scalar : !pto.tile_buf<...>, dtype) outs(%dst : !pto.tile_buf<...>)