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::BoolAttrbool attribute
regionId::mlir::IntegerAttr32-bit signless integer attribute whose value is non-negative
metric::mlir::triton::proton::MetricAttrallowed 32-bit signless integer cases: 0
granularity::mlir::triton::proton::GranularityAttrallowed 32-bit signless integer cases: 0, 1