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