ProtonOps

proton.record (triton::proton::RecordOp)

记录一个 GPU 硬件事件

语法

operation ::= `proton.record` `(` operands `)` attr-dict

该算子从性能计数器中记录 GPU 事件。目前仅支持周期计数器。

示例

proton.record() {isStart = true, regionId = 4 : i32}
...
proton.record() {isStart = false, regionId = 4 : i32}
...
proton.record() {isStart = true, regionId = 1 : i32, granularity = 1 : i32}
...
proton.record() {isStart = false, regionId = 1 : i32, granularity = 1 : i32}

接口: MemoryEffectOpInterface

属性:

属性	MLIR 类型	描述
isStart	::mlir::BoolAttr	布尔属性
regionId	::mlir::IntegerAttr	值为非负的 32 位无符号整数属性
metric	::mlir::triton::proton::MetricAttr	允许的 32 位无符号整数取值：0
granularity	::mlir::triton::proton::GranularityAttr	允许的 32 位无符号整数取值：0, 1