Skip to content

Commit 0a01c3a

Browse files
arsenmarsenm
committed
clang/AMDGPU: Emit atomicrmw from {global|flat}_atomic_fadd_v2f16 builtins
1 parent 0cdd4a2 commit 0a01c3a

File tree

4 files changed

+15
-19
lines changed

4 files changed

+15
-19
lines changed

‎clang/lib/CodeGen/CGBuiltin.cpp‎

Lines changed: 6 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -18920,22 +18920,15 @@ Value *CodeGenFunction::EmitAMDGPUBuiltinExpr(unsigned BuiltinID,
1892018920
Function *F = CGM.getIntrinsic(Intrin, { Src0->getType() });
1892118921
return Builder.CreateCall(F, { Src0, Builder.getFalse() });
1892218922
}
18923-
case AMDGPU::BI__builtin_amdgcn_global_atomic_fadd_v2f16:
1892418923
case AMDGPU::BI__builtin_amdgcn_global_atomic_fmin_f64:
1892518924
case AMDGPU::BI__builtin_amdgcn_global_atomic_fmax_f64:
1892618925
case AMDGPU::BI__builtin_amdgcn_flat_atomic_fadd_f64:
1892718926
case AMDGPU::BI__builtin_amdgcn_flat_atomic_fmin_f64:
1892818927
case AMDGPU::BI__builtin_amdgcn_flat_atomic_fmax_f64:
18929-
case AMDGPU::BI__builtin_amdgcn_flat_atomic_fadd_f32:
18930-
case AMDGPU::BI__builtin_amdgcn_flat_atomic_fadd_v2f16: {
18928+
case AMDGPU::BI__builtin_amdgcn_flat_atomic_fadd_f32: {
1893118929
Intrinsic::ID IID;
1893218930
llvm::Type *ArgTy = llvm::Type::getDoubleTy(getLLVMContext());
1893318931
switch (BuiltinID) {
18934-
case AMDGPU::BI__builtin_amdgcn_global_atomic_fadd_v2f16:
18935-
ArgTy = llvm::FixedVectorType::get(
18936-
llvm::Type::getHalfTy(getLLVMContext()), 2);
18937-
IID = Intrinsic::amdgcn_global_atomic_fadd;
18938-
break;
1893918932
case AMDGPU::BI__builtin_amdgcn_global_atomic_fmin_f64:
1894018933
IID = Intrinsic::amdgcn_global_atomic_fmin;
1894118934
break;
@@ -18955,11 +18948,6 @@ Value *CodeGenFunction::EmitAMDGPUBuiltinExpr(unsigned BuiltinID,
1895518948
ArgTy = llvm::Type::getFloatTy(getLLVMContext());
1895618949
IID = Intrinsic::amdgcn_flat_atomic_fadd;
1895718950
break;
18958-
case AMDGPU::BI__builtin_amdgcn_flat_atomic_fadd_v2f16:
18959-
ArgTy = llvm::FixedVectorType::get(
18960-
llvm::Type::getHalfTy(getLLVMContext()), 2);
18961-
IID = Intrinsic::amdgcn_flat_atomic_fadd;
18962-
break;
1896318951
}
1896418952
llvm::Value *Addr = EmitScalarExpr(E->getArg(0));
1896518953
llvm::Value *Val = EmitScalarExpr(E->getArg(1));
@@ -19360,7 +19348,9 @@ Value *CodeGenFunction::EmitAMDGPUBuiltinExpr(unsigned BuiltinID,
1936019348
case AMDGPU::BI__builtin_amdgcn_ds_fminf:
1936119349
case AMDGPU::BI__builtin_amdgcn_ds_fmaxf:
1936219350
case AMDGPU::BI__builtin_amdgcn_global_atomic_fadd_f32:
19363-
case AMDGPU::BI__builtin_amdgcn_global_atomic_fadd_f64: {
19351+
case AMDGPU::BI__builtin_amdgcn_global_atomic_fadd_f64:
19352+
case AMDGPU::BI__builtin_amdgcn_global_atomic_fadd_v2f16:
19353+
case AMDGPU::BI__builtin_amdgcn_flat_atomic_fadd_v2f16: {
1936419354
llvm::AtomicRMWInst::BinOp BinOp;
1936519355
switch (BuiltinID) {
1936619356
case AMDGPU::BI__builtin_amdgcn_atomic_inc32:
@@ -19378,6 +19368,8 @@ Value *CodeGenFunction::EmitAMDGPUBuiltinExpr(unsigned BuiltinID,
1937819368
case AMDGPU::BI__builtin_amdgcn_ds_atomic_fadd_v2bf16:
1937919369
case AMDGPU::BI__builtin_amdgcn_global_atomic_fadd_f32:
1938019370
case AMDGPU::BI__builtin_amdgcn_global_atomic_fadd_f64:
19371+
case AMDGPU::BI__builtin_amdgcn_global_atomic_fadd_v2f16:
19372+
case AMDGPU::BI__builtin_amdgcn_flat_atomic_fadd_v2f16:
1938119373
BinOp = llvm::AtomicRMWInst::FAdd;
1938219374
break;
1938319375
case AMDGPU::BI__builtin_amdgcn_ds_fminf:

‎clang/test/CodeGenOpenCL/builtins-fp-atomics-gfx12.cl‎

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,8 @@ void test_local_add_2f16_noret(__local half2 *addr, half2 x) {
4848
}
4949

5050
// CHECK-LABEL: test_flat_add_2f16
51-
// CHECK: call <2 x half> @llvm.amdgcn.flat.atomic.fadd.v2f16.p0.v2f16(ptr %{{.*}}, <2 x half> %{{.*}})
51+
// CHECK: [[RMW:%.+]] = atomicrmw fadd ptr %{{.+}}, <2 x half> %{{.+}} syncscope("agent") seq_cst, align 4, !amdgpu.no.fine.grained.memory !{{[0-9]+$}}
52+
5253
// GFX12-LABEL: test_flat_add_2f16
5354
// GFX12: flat_atomic_pk_add_f16
5455
half2 test_flat_add_2f16(__generic half2 *addr, half2 x) {
@@ -64,7 +65,8 @@ short2 test_flat_add_2bf16(__generic short2 *addr, short2 x) {
6465
}
6566

6667
// CHECK-LABEL: test_global_add_half2
67-
// CHECK: call <2 x half> @llvm.amdgcn.global.atomic.fadd.v2f16.p1.v2f16(ptr addrspace(1) %{{.*}}, <2 x half> %{{.*}})
68+
// CHECK: [[RMW:%.+]] = atomicrmw fadd ptr addrspace(1) %{{.+}}, <2 x half> %{{.+}} syncscope("agent") seq_cst, align 4, !amdgpu.no.fine.grained.memory !{{[0-9]+$}}
69+
6870
// GFX12-LABEL: test_global_add_half2
6971
// GFX12: global_atomic_pk_add_f16 v2, v[0:1], v2, off th:TH_ATOMIC_RETURN
7072
void test_global_add_half2(__global half2 *addr, half2 x) {
@@ -73,7 +75,8 @@ void test_global_add_half2(__global half2 *addr, half2 x) {
7375
}
7476

7577
// CHECK-LABEL: test_global_add_half2_noret
76-
// CHECK: call <2 x half> @llvm.amdgcn.global.atomic.fadd.v2f16.p1.v2f16(ptr addrspace(1) %{{.*}}, <2 x half> %{{.*}})
78+
// CHECK: [[RMW:%.+]] = atomicrmw fadd ptr addrspace(1) %{{.+}}, <2 x half> %{{.+}} syncscope("agent") seq_cst, align 4, !amdgpu.no.fine.grained.memory !{{[0-9]+$}}
79+
7780
// GFX12-LABEL: test_global_add_half2_noret
7881
// GFX12: global_atomic_pk_add_f16 v[0:1], v2, off
7982
void test_global_add_half2_noret(__global half2 *addr, half2 x) {

‎clang/test/CodeGenOpenCL/builtins-fp-atomics-gfx90a.cl‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ void test_global_add_f64(__global double *addr, double x) {
1818
}
1919

2020
// CHECK-LABEL: test_global_add_half2
21-
// CHECK: call <2 x half> @llvm.amdgcn.global.atomic.fadd.v2f16.p1.v2f16(ptr addrspace(1) %{{.*}}, <2 x half> %{{.*}})
21+
// CHECK: = atomicrmw fadd ptr addrspace(1) %{{.+}}, <2 x half> %{{.+}} syncscope("agent") seq_cst, align 4, !amdgpu.no.fine.grained.memory !{{[0-9]+$}}
2222
// GFX90A-LABEL: test_global_add_half2
2323
// GFX90A: global_atomic_pk_add_f16 v2, v[0:1], v2, off glc
2424
void test_global_add_half2(__global half2 *addr, half2 x) {

‎clang/test/CodeGenOpenCL/builtins-fp-atomics-gfx940.cl‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,8 @@ half2 test_flat_add_f32(__generic float *addr, float x) {
1818
}
1919

2020
// CHECK-LABEL: test_flat_add_2f16
21-
// CHECK: call <2 x half> @llvm.amdgcn.flat.atomic.fadd.v2f16.p0.v2f16(ptr %{{.*}}, <2 x half> %{{.*}})
21+
// CHECK: [[RMW:%.+]] = atomicrmw fadd ptr %{{.+}}, <2 x half> %{{.+}} syncscope("agent") seq_cst, align 4, !amdgpu.no.fine.grained.memory !{{[0-9]+$}}
22+
2223
// GFX940-LABEL: test_flat_add_2f16
2324
// GFX940: flat_atomic_pk_add_f16
2425
half2 test_flat_add_2f16(__generic half2 *addr, half2 x) {

0 commit comments

Comments
 (0)