Skip to content

Commit cfd79ea

Browse files
committed
Rename reductionPredicate to redcutionOperand
1 parent 4771142 commit cfd79ea

File tree

2 files changed

+11
-11
lines changed

2 files changed

+11
-11
lines changed

mlir/include/mlir/Dialect/LLVMIR/NVVMOps.td

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -973,7 +973,7 @@ def NVVM_BarrierOp : NVVM_Op<"barrier", [AttrSizedOperandSegments]> {
973973

974974
let arguments = (ins Optional<I32>:$barrierId, Optional<I32>:$numberOfThreads,
975975
OptionalAttr<BarrierReductionAttr>:$reductionOp,
976-
Optional<I32>:$reductionPredicate);
976+
Optional<I32>:$reductionOperand);
977977
string llvmBuilder = [{
978978
llvm::Value *id = $barrierId ? $barrierId : builder.getInt32(0);
979979
if ($numberOfThreads)
@@ -982,7 +982,7 @@ def NVVM_BarrierOp : NVVM_Op<"barrier", [AttrSizedOperandSegments]> {
982982
{id, $numberOfThreads});
983983
else if ($reductionOp)
984984
createIntrinsicCall(
985-
builder, getBarrierIntrinsicID($reductionOp), {$reductionPredicate});
985+
builder, getBarrierIntrinsicID($reductionOp), {$reductionOperand});
986986
else
987987
createIntrinsicCall(
988988
builder, llvm::Intrinsic::nvvm_barrier_cta_sync_aligned_all, {id});
@@ -993,7 +993,7 @@ def NVVM_BarrierOp : NVVM_Op<"barrier", [AttrSizedOperandSegments]> {
993993

994994
let assemblyFormat =
995995
"(`id` `=` $barrierId^)? (`number_of_threads` `=` $numberOfThreads^)? "
996-
"($reductionOp^ $reductionPredicate)? (`->` type($res)^)? attr-dict";
996+
"($reductionOp^ $reductionOperand)? (`->` type($res)^)? attr-dict";
997997

998998
let builders = [OpBuilder<(ins), [{
999999
return build($_builder, $_state, TypeRange{}, Value{}, Value{}, {}, Value{});
Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,20 +1,20 @@
11
// RUN: mlir-translate -mlir-to-llvmir %s -split-input-file --verify-diagnostics | FileCheck %s
22

33
// CHECK-LABEL: @llvm_nvvm_barrier(
4-
// CHECK-SAME: i32 %[[barId:.*]], i32 %[[numThreads:.*]], i32 %[[predicate:.*]])
5-
llvm.func @llvm_nvvm_barrier(%barID : i32, %numberOfThreads : i32, %predicate : i32) {
4+
// CHECK-SAME: i32 %[[barId:.*]], i32 %[[numThreads:.*]], i32 %[[redOperand:.*]])
5+
llvm.func @llvm_nvvm_barrier(%barID : i32, %numberOfThreads : i32, %redOperand : i32) {
66
// CHECK: call void @llvm.nvvm.barrier.cta.sync.aligned.all(i32 0)
77
nvvm.barrier
88
// CHECK: call void @llvm.nvvm.barrier.cta.sync.aligned.all(i32 %[[barId]])
99
nvvm.barrier id = %barID
1010
// CHECK: call void @llvm.nvvm.barrier.cta.sync.aligned.count(i32 %[[barId]], i32 %[[numThreads]])
1111
nvvm.barrier id = %barID number_of_threads = %numberOfThreads
12-
// CHECK: %{{.*}} = call i32 @llvm.nvvm.barrier0.and(i32 %[[predicate]])
13-
%0 = nvvm.barrier #nvvm.reduction<and> %predicate -> i32
14-
// CHECK: %{{.*}} = call i32 @llvm.nvvm.barrier0.or(i32 %[[predicate]])
15-
%1 = nvvm.barrier #nvvm.reduction<or> %predicate -> i32
16-
// CHECK: %{{.*}} = call i32 @llvm.nvvm.barrier0.popc(i32 %[[predicate]])
17-
%2 = nvvm.barrier #nvvm.reduction<popc> %predicate -> i32
12+
// CHECK: %{{.*}} = call i32 @llvm.nvvm.barrier0.and(i32 %[[redOperand]])
13+
%0 = nvvm.barrier #nvvm.reduction<and> %redOperand -> i32
14+
// CHECK: %{{.*}} = call i32 @llvm.nvvm.barrier0.or(i32 %[[redOperand]])
15+
%1 = nvvm.barrier #nvvm.reduction<or> %redOperand -> i32
16+
// CHECK: %{{.*}} = call i32 @llvm.nvvm.barrier0.popc(i32 %[[redOperand]])
17+
%2 = nvvm.barrier #nvvm.reduction<popc> %redOperand -> i32
1818

1919
llvm.return
2020
}

0 commit comments

Comments
 (0)