fdo-mirrors/mesa
1
0
Fork 0
mirror of https://gitlab.freedesktop.org/mesa/mesa.git synced 2026-05-09 04:38:03 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

ac/llvm: Use float types for float atomics

Browse source 
Reviewed-by: Marek Olšák <marek.olsak@amd.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/25010>
This commit is contained in:
Konstantin Seurer 2023-09-02 14:24:37 +02:00 • committed by Marge Bot
parent 60e7b1c69c
commit 4c168635f8
1 changed files with 19 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
src/amd/llvm/ac_nir_to_llvm.c
View file

@ -2080,6 +2080,22 @@ static LLVMValueRef visit_global_atomic(struct ac_nir_context *ctx,
LLVMTypeRef data_type = LLVMTypeOf(data);
assert(instr->src[1].ssa->num_components == 1);
if (is_float) {
switch (instr->src[1].ssa->bit_size) {
case 32:
data_type = ctx->ac.f32;
break;
case 64:
data_type = ctx->ac.f64;
break;
default:
unreachable("Unsupported float bit size");
}
data = LLVMBuildBitCast(ctx->ac.builder, data, data_type, "");
}
LLVMValueRef addr = get_global_address(ctx, instr, data_type);
if (instr->intrinsic == nir_intrinsic_global_atomic_swap ||
@ -2097,15 +2113,16 @@ static LLVMValueRef visit_global_atomic(struct ac_nir_context *ctx,
params[arg_count++] = data;
ac_build_type_name_for_intr(data_type, type, sizeof(type));
snprintf(name, sizeof(name), "llvm.amdgcn.global.atomic.%s.%s.p1%s.%s", op, type, type, type);
snprintf(name, sizeof(name), "llvm.amdgcn.global.atomic.%s.%s.p1.%s", op, type, type);
result = ac_build_intrinsic(&ctx->ac, name, data_type, params, arg_count, 0);
result = ac_to_integer(&ctx->ac, result);
} else {
op = translate_atomic_op(nir_op);
result = ac_build_atomic_rmw(&ctx->ac, op, addr, ac_to_integer(&ctx->ac, data), sync_scope);
}
result = ac_to_integer(&ctx->ac, result);
return result;
}