OpenHarmony/third_party_mesa3d
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ac/nir: add support for nir_intrinsic_shared_atomic_fadd

Browse Source 
Only LLVM 10+ has support.

Signed-off-by: Samuel Pitoiset <samuel.pitoiset@gmail.com>
Reviewed-by: Daniel Schürmann <daniel@schuermann.dev>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/6000>
This commit is contained in:
Samuel Pitoiset
2020-07-21 10:49:36 +02:00
parent 7615f2d690
commit b8517e5ef9
1 changed files with 20 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/amd/llvm/ac_nir_to_llvm.c
View File

@@ -3404,11 +3404,26 @@ static LLVMValueRef visit_var_atomic(struct ac_nir_context *ctx,
case nir_intrinsic_deref_atomic_exchange: case nir_intrinsic_deref_atomic_exchange:
op = LLVMAtomicRMWBinOpXchg; op = LLVMAtomicRMWBinOpXchg;
break; break;
#if LLVM_VERSION_MAJOR >= 10
case nir_intrinsic_shared_atomic_fadd:
case nir_intrinsic_deref_atomic_fadd:
op = LLVMAtomicRMWBinOpFAdd;
break;
#endif
default: default:
return NULL; return NULL;
} }
result = ac_build_atomic_rmw(&ctx->ac, op, ptr, ac_to_integer(&ctx->ac, src), sync_scope); LLVMValueRef val;
if (instr->intrinsic == nir_intrinsic_shared_atomic_fadd ||
instr->intrinsic == nir_intrinsic_deref_atomic_fadd) {
val = ac_to_float(&ctx->ac, src);
} else {
val = ac_to_integer(&ctx->ac, src);
}
result = ac_build_atomic_rmw(&ctx->ac, op, ptr, val, sync_scope);
} }
if (ctx->ac.postponed_kill) if (ctx->ac.postponed_kill)
@@ -3944,7 +3959,8 @@ static void visit_intrinsic(struct ac_nir_context *ctx,
case nir_intrinsic_shared_atomic_or: case nir_intrinsic_shared_atomic_or:
case nir_intrinsic_shared_atomic_xor: case nir_intrinsic_shared_atomic_xor:
case nir_intrinsic_shared_atomic_exchange: case nir_intrinsic_shared_atomic_exchange:
case nir_intrinsic_shared_atomic_comp_swap: { case nir_intrinsic_shared_atomic_comp_swap:
case nir_intrinsic_shared_atomic_fadd: {
LLVMValueRef ptr = get_memory_ptr(ctx, instr->src[0], LLVMValueRef ptr = get_memory_ptr(ctx, instr->src[0],
instr->src[1].ssa->bit_size); instr->src[1].ssa->bit_size);
result = visit_var_atomic(ctx, instr, ptr, 1); result = visit_var_atomic(ctx, instr, ptr, 1);
@@ -3959,7 +3975,8 @@ static void visit_intrinsic(struct ac_nir_context *ctx,
case nir_intrinsic_deref_atomic_or: case nir_intrinsic_deref_atomic_or:
case nir_intrinsic_deref_atomic_xor: case nir_intrinsic_deref_atomic_xor:
case nir_intrinsic_deref_atomic_exchange: case nir_intrinsic_deref_atomic_exchange:
case nir_intrinsic_deref_atomic_comp_swap: { case nir_intrinsic_deref_atomic_comp_swap:
case nir_intrinsic_deref_atomic_fadd: {
LLVMValueRef ptr = get_src(ctx, instr->src[0]); LLVMValueRef ptr = get_src(ctx, instr->src[0]);
result = visit_var_atomic(ctx, instr, ptr, 1); result = visit_var_atomic(ctx, instr, ptr, 1);
break; break;