[MLIR][NVVM] Add default constructor for nvvm.barrier [NFC] (#126225)

This PR adds a default constructor to `nvvm.barrier`, making it more
convenient to build the OP.
This commit is contained in:
Guray Ozen 2025-02-07 15:42:57 +01:00 committed by GitHub
parent 1f2c36a879
commit b284a849d5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -463,7 +463,17 @@ def NVVM_BarrierOp : NVVM_Op<"barrier", [AttrSizedOperandSegments]> {
}
}];
let hasVerifier = 1;
let assemblyFormat = "(`id` `=` $barrierId^)? (`number_of_threads` `=` $numberOfThreads^)? attr-dict";
let builders = [
OpBuilder<(ins), [{
return build($_builder, $_state, Value{}, Value{});
}]>,
OpBuilder<(ins "Value":$barrierId), [{
return build($_builder, $_state, barrierId, Value{});
}]>
];
}
def NVVM_BarrierArriveOp : NVVM_PTXBuilder_Op<"barrier.arrive">