2022-04-23 09:59:50 +01:00
|
|
|
// SPDX-FileCopyrightText: Copyright 2021 yuzu Emulator Project
|
|
|
|
// SPDX-License-Identifier: GPL-2.0-or-later
|
2021-04-11 07:07:02 +01:00
|
|
|
|
2022-03-22 21:11:24 +00:00
|
|
|
#include <bit>
|
|
|
|
|
2021-04-11 07:07:02 +01:00
|
|
|
#include "shader_recompiler/backend/spirv/emit_spirv.h"
|
2021-05-04 00:53:00 +01:00
|
|
|
#include "shader_recompiler/backend/spirv/emit_spirv_instructions.h"
|
2021-12-05 22:24:54 +00:00
|
|
|
#include "shader_recompiler/backend/spirv/spirv_emit_context.h"
|
2021-04-11 07:07:02 +01:00
|
|
|
|
|
|
|
namespace Shader::Backend::SPIRV {
|
|
|
|
namespace {
|
2021-04-13 09:32:21 +01:00
|
|
|
Id SharedPointer(EmitContext& ctx, Id offset, u32 index_offset = 0) {
|
2021-04-19 01:47:31 +01:00
|
|
|
const Id shift_id{ctx.Const(2U)};
|
2021-04-13 09:32:21 +01:00
|
|
|
Id index{ctx.OpShiftRightArithmetic(ctx.U32[1], offset, shift_id)};
|
|
|
|
if (index_offset > 0) {
|
2021-04-19 01:47:31 +01:00
|
|
|
index = ctx.OpIAdd(ctx.U32[1], index, ctx.Const(index_offset));
|
2021-04-13 09:32:21 +01:00
|
|
|
}
|
2021-04-11 07:07:02 +01:00
|
|
|
return ctx.profile.support_explicit_workgroup_layout
|
|
|
|
? ctx.OpAccessChain(ctx.shared_u32, ctx.shared_memory_u32, ctx.u32_zero_value, index)
|
|
|
|
: ctx.OpAccessChain(ctx.shared_u32, ctx.shared_memory_u32, index);
|
|
|
|
}
|
|
|
|
|
|
|
|
Id StorageIndex(EmitContext& ctx, const IR::Value& offset, size_t element_size) {
|
|
|
|
if (offset.IsImmediate()) {
|
|
|
|
const u32 imm_offset{static_cast<u32>(offset.U32() / element_size)};
|
2021-04-19 01:47:31 +01:00
|
|
|
return ctx.Const(imm_offset);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
const u32 shift{static_cast<u32>(std::countr_zero(element_size))};
|
|
|
|
const Id index{ctx.Def(offset)};
|
|
|
|
if (shift == 0) {
|
|
|
|
return index;
|
|
|
|
}
|
2021-04-19 01:47:31 +01:00
|
|
|
const Id shift_id{ctx.Const(shift)};
|
2021-04-11 07:07:02 +01:00
|
|
|
return ctx.OpShiftRightLogical(ctx.U32[1], index, shift_id);
|
|
|
|
}
|
|
|
|
|
2021-04-13 09:32:21 +01:00
|
|
|
Id StoragePointer(EmitContext& ctx, const StorageTypeDefinition& type_def,
|
|
|
|
Id StorageDefinitions::*member_ptr, const IR::Value& binding,
|
|
|
|
const IR::Value& offset, size_t element_size) {
|
2021-04-11 07:07:02 +01:00
|
|
|
if (!binding.IsImmediate()) {
|
|
|
|
throw NotImplementedException("Dynamic storage buffer indexing");
|
|
|
|
}
|
2021-04-13 09:32:21 +01:00
|
|
|
const Id ssbo{ctx.ssbos[binding.U32()].*member_ptr};
|
|
|
|
const Id index{StorageIndex(ctx, offset, element_size)};
|
|
|
|
return ctx.OpAccessChain(type_def.element, ssbo, ctx.u32_zero_value, index);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
2021-04-13 09:32:21 +01:00
|
|
|
std::pair<Id, Id> AtomicArgs(EmitContext& ctx) {
|
2021-04-19 01:47:31 +01:00
|
|
|
const Id scope{ctx.Const(static_cast<u32>(spv::Scope::Device))};
|
2021-04-11 07:07:02 +01:00
|
|
|
const Id semantics{ctx.u32_zero_value};
|
|
|
|
return {scope, semantics};
|
|
|
|
}
|
|
|
|
|
2021-04-13 09:32:21 +01:00
|
|
|
Id SharedAtomicU32(EmitContext& ctx, Id offset, Id value,
|
|
|
|
Id (Sirit::Module::*atomic_func)(Id, Id, Id, Id, Id)) {
|
|
|
|
const Id pointer{SharedPointer(ctx, offset)};
|
|
|
|
const auto [scope, semantics]{AtomicArgs(ctx)};
|
|
|
|
return (ctx.*atomic_func)(ctx.U32[1], pointer, scope, semantics, value);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
2021-04-13 09:32:21 +01:00
|
|
|
Id StorageAtomicU32(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset, Id value,
|
|
|
|
Id (Sirit::Module::*atomic_func)(Id, Id, Id, Id, Id)) {
|
|
|
|
const Id pointer{StoragePointer(ctx, ctx.storage_types.U32, &StorageDefinitions::U32, binding,
|
|
|
|
offset, sizeof(u32))};
|
|
|
|
const auto [scope, semantics]{AtomicArgs(ctx)};
|
|
|
|
return (ctx.*atomic_func)(ctx.U32[1], pointer, scope, semantics, value);
|
|
|
|
}
|
|
|
|
|
|
|
|
Id StorageAtomicU64(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset, Id value,
|
|
|
|
Id (Sirit::Module::*atomic_func)(Id, Id, Id, Id, Id),
|
|
|
|
Id (Sirit::Module::*non_atomic_func)(Id, Id, Id)) {
|
2023-05-03 01:52:41 +01:00
|
|
|
if (!ctx.profile.support_descriptor_aliasing) {
|
|
|
|
LOG_WARNING(Shader_SPIRV, "Descriptor aliasing not supported, this cannot be atomic.");
|
|
|
|
return ctx.ConstantNull(ctx.U64);
|
|
|
|
}
|
|
|
|
|
2021-04-13 09:32:21 +01:00
|
|
|
if (ctx.profile.support_int64_atomics) {
|
|
|
|
const Id pointer{StoragePointer(ctx, ctx.storage_types.U64, &StorageDefinitions::U64,
|
|
|
|
binding, offset, sizeof(u64))};
|
|
|
|
const auto [scope, semantics]{AtomicArgs(ctx)};
|
|
|
|
return (ctx.*atomic_func)(ctx.U64, pointer, scope, semantics, value);
|
|
|
|
}
|
2022-01-29 00:00:04 +00:00
|
|
|
LOG_WARNING(Shader_SPIRV, "Int64 atomics not supported, fallback to non-atomic");
|
2021-04-13 09:32:21 +01:00
|
|
|
const Id pointer{StoragePointer(ctx, ctx.storage_types.U32x2, &StorageDefinitions::U32x2,
|
|
|
|
binding, offset, sizeof(u32[2]))};
|
|
|
|
const Id original_value{ctx.OpBitcast(ctx.U64, ctx.OpLoad(ctx.U32[2], pointer))};
|
|
|
|
const Id result{(ctx.*non_atomic_func)(ctx.U64, value, original_value)};
|
2021-04-14 06:17:47 +01:00
|
|
|
ctx.OpStore(pointer, ctx.OpBitcast(ctx.U32[2], result));
|
2021-04-13 09:32:21 +01:00
|
|
|
return original_value;
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
2022-01-29 18:46:06 +00:00
|
|
|
|
|
|
|
Id StorageAtomicU32x2(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset, Id value,
|
|
|
|
Id (Sirit::Module::*non_atomic_func)(Id, Id, Id)) {
|
2023-05-03 01:52:41 +01:00
|
|
|
if (!ctx.profile.support_descriptor_aliasing) {
|
|
|
|
LOG_WARNING(Shader_SPIRV, "Descriptor aliasing not supported, this cannot be atomic.");
|
|
|
|
return ctx.ConstantNull(ctx.U32[2]);
|
|
|
|
}
|
|
|
|
|
2022-01-29 18:46:06 +00:00
|
|
|
LOG_WARNING(Shader_SPIRV, "Int64 atomics not supported, fallback to non-atomic");
|
|
|
|
const Id pointer{StoragePointer(ctx, ctx.storage_types.U32x2, &StorageDefinitions::U32x2,
|
|
|
|
binding, offset, sizeof(u32[2]))};
|
|
|
|
const Id original_value{ctx.OpLoad(ctx.U32[2], pointer)};
|
|
|
|
const Id result{(ctx.*non_atomic_func)(ctx.U32[2], value, original_value)};
|
|
|
|
ctx.OpStore(pointer, result);
|
|
|
|
return original_value;
|
|
|
|
}
|
2021-04-11 07:07:02 +01:00
|
|
|
} // Anonymous namespace
|
|
|
|
|
2021-04-13 09:32:21 +01:00
|
|
|
Id EmitSharedAtomicIAdd32(EmitContext& ctx, Id offset, Id value) {
|
|
|
|
return SharedAtomicU32(ctx, offset, value, &Sirit::Module::OpAtomicIAdd);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
2021-04-13 09:32:21 +01:00
|
|
|
Id EmitSharedAtomicSMin32(EmitContext& ctx, Id offset, Id value) {
|
|
|
|
return SharedAtomicU32(ctx, offset, value, &Sirit::Module::OpAtomicSMin);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
2021-04-13 09:32:21 +01:00
|
|
|
Id EmitSharedAtomicUMin32(EmitContext& ctx, Id offset, Id value) {
|
|
|
|
return SharedAtomicU32(ctx, offset, value, &Sirit::Module::OpAtomicUMin);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
2021-04-13 09:32:21 +01:00
|
|
|
Id EmitSharedAtomicSMax32(EmitContext& ctx, Id offset, Id value) {
|
|
|
|
return SharedAtomicU32(ctx, offset, value, &Sirit::Module::OpAtomicSMax);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
2021-04-13 09:32:21 +01:00
|
|
|
Id EmitSharedAtomicUMax32(EmitContext& ctx, Id offset, Id value) {
|
|
|
|
return SharedAtomicU32(ctx, offset, value, &Sirit::Module::OpAtomicUMax);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
2021-04-13 09:32:21 +01:00
|
|
|
Id EmitSharedAtomicInc32(EmitContext& ctx, Id offset, Id value) {
|
2021-04-19 01:47:31 +01:00
|
|
|
const Id shift_id{ctx.Const(2U)};
|
2021-04-13 09:32:21 +01:00
|
|
|
const Id index{ctx.OpShiftRightArithmetic(ctx.U32[1], offset, shift_id)};
|
|
|
|
return ctx.OpFunctionCall(ctx.U32[1], ctx.increment_cas_shared, index, value);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
2021-04-13 09:32:21 +01:00
|
|
|
Id EmitSharedAtomicDec32(EmitContext& ctx, Id offset, Id value) {
|
2021-04-19 01:47:31 +01:00
|
|
|
const Id shift_id{ctx.Const(2U)};
|
2021-04-13 09:32:21 +01:00
|
|
|
const Id index{ctx.OpShiftRightArithmetic(ctx.U32[1], offset, shift_id)};
|
|
|
|
return ctx.OpFunctionCall(ctx.U32[1], ctx.decrement_cas_shared, index, value);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
2021-04-13 09:32:21 +01:00
|
|
|
Id EmitSharedAtomicAnd32(EmitContext& ctx, Id offset, Id value) {
|
|
|
|
return SharedAtomicU32(ctx, offset, value, &Sirit::Module::OpAtomicAnd);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
2021-04-13 09:32:21 +01:00
|
|
|
Id EmitSharedAtomicOr32(EmitContext& ctx, Id offset, Id value) {
|
|
|
|
return SharedAtomicU32(ctx, offset, value, &Sirit::Module::OpAtomicOr);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
2021-04-13 09:32:21 +01:00
|
|
|
Id EmitSharedAtomicXor32(EmitContext& ctx, Id offset, Id value) {
|
|
|
|
return SharedAtomicU32(ctx, offset, value, &Sirit::Module::OpAtomicXor);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
2021-04-13 09:32:21 +01:00
|
|
|
Id EmitSharedAtomicExchange32(EmitContext& ctx, Id offset, Id value) {
|
|
|
|
return SharedAtomicU32(ctx, offset, value, &Sirit::Module::OpAtomicExchange);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
2021-04-13 09:32:21 +01:00
|
|
|
Id EmitSharedAtomicExchange64(EmitContext& ctx, Id offset, Id value) {
|
|
|
|
if (ctx.profile.support_int64_atomics && ctx.profile.support_explicit_workgroup_layout) {
|
2021-04-19 01:47:31 +01:00
|
|
|
const Id shift_id{ctx.Const(3U)};
|
2021-04-13 09:32:21 +01:00
|
|
|
const Id index{ctx.OpShiftRightArithmetic(ctx.U32[1], offset, shift_id)};
|
|
|
|
const Id pointer{
|
|
|
|
ctx.OpAccessChain(ctx.shared_u64, ctx.shared_memory_u64, ctx.u32_zero_value, index)};
|
|
|
|
const auto [scope, semantics]{AtomicArgs(ctx)};
|
|
|
|
return ctx.OpAtomicExchange(ctx.U64, pointer, scope, semantics, value);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
2022-01-29 18:46:06 +00:00
|
|
|
LOG_WARNING(Shader_SPIRV, "Int64 atomics not supported, fallback to non-atomic");
|
2021-04-13 09:32:21 +01:00
|
|
|
const Id pointer_1{SharedPointer(ctx, offset, 0)};
|
|
|
|
const Id pointer_2{SharedPointer(ctx, offset, 1)};
|
|
|
|
const Id value_1{ctx.OpLoad(ctx.U32[1], pointer_1)};
|
|
|
|
const Id value_2{ctx.OpLoad(ctx.U32[1], pointer_2)};
|
|
|
|
const Id new_vector{ctx.OpBitcast(ctx.U32[2], value)};
|
|
|
|
ctx.OpStore(pointer_1, ctx.OpCompositeExtract(ctx.U32[1], new_vector, 0U));
|
|
|
|
ctx.OpStore(pointer_2, ctx.OpCompositeExtract(ctx.U32[1], new_vector, 1U));
|
|
|
|
return ctx.OpBitcast(ctx.U64, ctx.OpCompositeConstruct(ctx.U32[2], value_1, value_2));
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
2022-01-29 18:46:06 +00:00
|
|
|
Id EmitSharedAtomicExchange32x2(EmitContext& ctx, Id offset, Id value) {
|
|
|
|
LOG_WARNING(Shader_SPIRV, "Int64 atomics not supported, fallback to non-atomic");
|
|
|
|
const Id pointer_1{SharedPointer(ctx, offset, 0)};
|
|
|
|
const Id pointer_2{SharedPointer(ctx, offset, 1)};
|
|
|
|
const Id value_1{ctx.OpLoad(ctx.U32[1], pointer_1)};
|
|
|
|
const Id value_2{ctx.OpLoad(ctx.U32[1], pointer_2)};
|
|
|
|
const Id new_vector{ctx.OpBitcast(ctx.U32[2], value)};
|
|
|
|
ctx.OpStore(pointer_1, ctx.OpCompositeExtract(ctx.U32[1], new_vector, 0U));
|
|
|
|
ctx.OpStore(pointer_2, ctx.OpCompositeExtract(ctx.U32[1], new_vector, 1U));
|
|
|
|
return ctx.OpCompositeConstruct(ctx.U32[2], value_1, value_2);
|
|
|
|
}
|
|
|
|
|
2021-04-11 07:07:02 +01:00
|
|
|
Id EmitStorageAtomicIAdd32(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
return StorageAtomicU32(ctx, binding, offset, value, &Sirit::Module::OpAtomicIAdd);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicSMin32(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
return StorageAtomicU32(ctx, binding, offset, value, &Sirit::Module::OpAtomicSMin);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicUMin32(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
return StorageAtomicU32(ctx, binding, offset, value, &Sirit::Module::OpAtomicUMin);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicSMax32(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
return StorageAtomicU32(ctx, binding, offset, value, &Sirit::Module::OpAtomicSMax);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicUMax32(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
return StorageAtomicU32(ctx, binding, offset, value, &Sirit::Module::OpAtomicUMax);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicInc32(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
const Id ssbo{ctx.ssbos[binding.U32()].U32};
|
2021-04-11 07:07:02 +01:00
|
|
|
const Id base_index{StorageIndex(ctx, offset, sizeof(u32))};
|
|
|
|
return ctx.OpFunctionCall(ctx.U32[1], ctx.increment_cas_ssbo, base_index, value, ssbo);
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicDec32(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
const Id ssbo{ctx.ssbos[binding.U32()].U32};
|
2021-04-11 07:07:02 +01:00
|
|
|
const Id base_index{StorageIndex(ctx, offset, sizeof(u32))};
|
|
|
|
return ctx.OpFunctionCall(ctx.U32[1], ctx.decrement_cas_ssbo, base_index, value, ssbo);
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicAnd32(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
return StorageAtomicU32(ctx, binding, offset, value, &Sirit::Module::OpAtomicAnd);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicOr32(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
return StorageAtomicU32(ctx, binding, offset, value, &Sirit::Module::OpAtomicOr);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicXor32(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
return StorageAtomicU32(ctx, binding, offset, value, &Sirit::Module::OpAtomicXor);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicExchange32(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
return StorageAtomicU32(ctx, binding, offset, value, &Sirit::Module::OpAtomicExchange);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicIAdd64(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
return StorageAtomicU64(ctx, binding, offset, value, &Sirit::Module::OpAtomicIAdd,
|
|
|
|
&Sirit::Module::OpIAdd);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicSMin64(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
return StorageAtomicU64(ctx, binding, offset, value, &Sirit::Module::OpAtomicSMin,
|
|
|
|
&Sirit::Module::OpSMin);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicUMin64(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
return StorageAtomicU64(ctx, binding, offset, value, &Sirit::Module::OpAtomicUMin,
|
|
|
|
&Sirit::Module::OpUMin);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicSMax64(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
return StorageAtomicU64(ctx, binding, offset, value, &Sirit::Module::OpAtomicSMax,
|
|
|
|
&Sirit::Module::OpSMax);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicUMax64(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
return StorageAtomicU64(ctx, binding, offset, value, &Sirit::Module::OpAtomicUMax,
|
|
|
|
&Sirit::Module::OpUMax);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicAnd64(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
return StorageAtomicU64(ctx, binding, offset, value, &Sirit::Module::OpAtomicAnd,
|
|
|
|
&Sirit::Module::OpBitwiseAnd);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicOr64(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
return StorageAtomicU64(ctx, binding, offset, value, &Sirit::Module::OpAtomicOr,
|
|
|
|
&Sirit::Module::OpBitwiseOr);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicXor64(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
return StorageAtomicU64(ctx, binding, offset, value, &Sirit::Module::OpAtomicXor,
|
|
|
|
&Sirit::Module::OpBitwiseXor);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicExchange64(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
|
|
|
if (ctx.profile.support_int64_atomics) {
|
2021-04-13 09:32:21 +01:00
|
|
|
const Id pointer{StoragePointer(ctx, ctx.storage_types.U64, &StorageDefinitions::U64,
|
|
|
|
binding, offset, sizeof(u64))};
|
|
|
|
const auto [scope, semantics]{AtomicArgs(ctx)};
|
|
|
|
return ctx.OpAtomicExchange(ctx.U64, pointer, scope, semantics, value);
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
2022-01-29 00:00:04 +00:00
|
|
|
LOG_WARNING(Shader_SPIRV, "Int64 atomics not supported, fallback to non-atomic");
|
2021-04-13 09:32:21 +01:00
|
|
|
const Id pointer{StoragePointer(ctx, ctx.storage_types.U32x2, &StorageDefinitions::U32x2,
|
|
|
|
binding, offset, sizeof(u32[2]))};
|
|
|
|
const Id original{ctx.OpBitcast(ctx.U64, ctx.OpLoad(ctx.U32[2], pointer))};
|
|
|
|
ctx.OpStore(pointer, value);
|
|
|
|
return original;
|
2021-04-11 07:07:02 +01:00
|
|
|
}
|
|
|
|
|
2022-01-29 18:46:06 +00:00
|
|
|
Id EmitStorageAtomicIAdd32x2(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
|
|
|
return StorageAtomicU32x2(ctx, binding, offset, value, &Sirit::Module::OpIAdd);
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicSMin32x2(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
|
|
|
return StorageAtomicU32x2(ctx, binding, offset, value, &Sirit::Module::OpSMin);
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicUMin32x2(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
|
|
|
return StorageAtomicU32x2(ctx, binding, offset, value, &Sirit::Module::OpUMin);
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicSMax32x2(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
|
|
|
return StorageAtomicU32x2(ctx, binding, offset, value, &Sirit::Module::OpSMax);
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicUMax32x2(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
|
|
|
return StorageAtomicU32x2(ctx, binding, offset, value, &Sirit::Module::OpUMax);
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicAnd32x2(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
|
|
|
return StorageAtomicU32x2(ctx, binding, offset, value, &Sirit::Module::OpBitwiseAnd);
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicOr32x2(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
|
|
|
return StorageAtomicU32x2(ctx, binding, offset, value, &Sirit::Module::OpBitwiseOr);
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicXor32x2(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
|
|
|
return StorageAtomicU32x2(ctx, binding, offset, value, &Sirit::Module::OpBitwiseXor);
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicExchange32x2(EmitContext& ctx, const IR::Value& binding,
|
|
|
|
const IR::Value& offset, Id value) {
|
|
|
|
LOG_WARNING(Shader_SPIRV, "Int64 atomics not supported, fallback to non-atomic");
|
|
|
|
const Id pointer{StoragePointer(ctx, ctx.storage_types.U32x2, &StorageDefinitions::U32x2,
|
|
|
|
binding, offset, sizeof(u32[2]))};
|
|
|
|
const Id original{ctx.OpLoad(ctx.U32[2], pointer)};
|
|
|
|
ctx.OpStore(pointer, value);
|
|
|
|
return original;
|
|
|
|
}
|
|
|
|
|
2021-04-11 07:07:02 +01:00
|
|
|
Id EmitStorageAtomicAddF32(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
const Id ssbo{ctx.ssbos[binding.U32()].U32};
|
2021-04-11 07:07:02 +01:00
|
|
|
const Id base_index{StorageIndex(ctx, offset, sizeof(u32))};
|
|
|
|
return ctx.OpFunctionCall(ctx.F32[1], ctx.f32_add_cas, base_index, value, ssbo);
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicAddF16x2(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
const Id ssbo{ctx.ssbos[binding.U32()].U32};
|
2021-04-11 07:07:02 +01:00
|
|
|
const Id base_index{StorageIndex(ctx, offset, sizeof(u32))};
|
|
|
|
const Id result{ctx.OpFunctionCall(ctx.F16[2], ctx.f16x2_add_cas, base_index, value, ssbo)};
|
|
|
|
return ctx.OpBitcast(ctx.U32[1], result);
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicAddF32x2(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
const Id ssbo{ctx.ssbos[binding.U32()].U32};
|
2021-04-11 07:07:02 +01:00
|
|
|
const Id base_index{StorageIndex(ctx, offset, sizeof(u32))};
|
|
|
|
const Id result{ctx.OpFunctionCall(ctx.F32[2], ctx.f32x2_add_cas, base_index, value, ssbo)};
|
|
|
|
return ctx.OpPackHalf2x16(ctx.U32[1], result);
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicMinF16x2(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
const Id ssbo{ctx.ssbos[binding.U32()].U32};
|
2021-04-11 07:07:02 +01:00
|
|
|
const Id base_index{StorageIndex(ctx, offset, sizeof(u32))};
|
|
|
|
const Id result{ctx.OpFunctionCall(ctx.F16[2], ctx.f16x2_min_cas, base_index, value, ssbo)};
|
|
|
|
return ctx.OpBitcast(ctx.U32[1], result);
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicMinF32x2(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
const Id ssbo{ctx.ssbos[binding.U32()].U32};
|
2021-04-11 07:07:02 +01:00
|
|
|
const Id base_index{StorageIndex(ctx, offset, sizeof(u32))};
|
|
|
|
const Id result{ctx.OpFunctionCall(ctx.F32[2], ctx.f32x2_min_cas, base_index, value, ssbo)};
|
|
|
|
return ctx.OpPackHalf2x16(ctx.U32[1], result);
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicMaxF16x2(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
const Id ssbo{ctx.ssbos[binding.U32()].U32};
|
2021-04-11 07:07:02 +01:00
|
|
|
const Id base_index{StorageIndex(ctx, offset, sizeof(u32))};
|
|
|
|
const Id result{ctx.OpFunctionCall(ctx.F16[2], ctx.f16x2_max_cas, base_index, value, ssbo)};
|
|
|
|
return ctx.OpBitcast(ctx.U32[1], result);
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitStorageAtomicMaxF32x2(EmitContext& ctx, const IR::Value& binding, const IR::Value& offset,
|
|
|
|
Id value) {
|
2021-04-13 09:32:21 +01:00
|
|
|
const Id ssbo{ctx.ssbos[binding.U32()].U32};
|
2021-04-11 07:07:02 +01:00
|
|
|
const Id base_index{StorageIndex(ctx, offset, sizeof(u32))};
|
|
|
|
const Id result{ctx.OpFunctionCall(ctx.F32[2], ctx.f32x2_max_cas, base_index, value, ssbo)};
|
|
|
|
return ctx.OpPackHalf2x16(ctx.U32[1], result);
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicIAdd32(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicSMin32(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicUMin32(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicSMax32(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicUMax32(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicInc32(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicDec32(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicAnd32(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicOr32(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicXor32(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicExchange32(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicIAdd64(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicSMin64(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicUMin64(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicSMax64(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicUMax64(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicInc64(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicDec64(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicAnd64(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicOr64(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicXor64(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicExchange64(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
2022-01-29 18:46:06 +00:00
|
|
|
Id EmitGlobalAtomicIAdd32x2(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicSMin32x2(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicUMin32x2(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicSMax32x2(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicUMax32x2(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicInc32x2(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicDec32x2(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicAnd32x2(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicOr32x2(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicXor32x2(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicExchange32x2(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
2021-04-11 07:07:02 +01:00
|
|
|
Id EmitGlobalAtomicAddF32(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicAddF16x2(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicAddF32x2(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicMinF16x2(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicMinF32x2(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicMaxF16x2(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
Id EmitGlobalAtomicMaxF32x2(EmitContext&) {
|
|
|
|
throw NotImplementedException("SPIR-V Instruction");
|
|
|
|
}
|
|
|
|
|
|
|
|
} // namespace Shader::Backend::SPIRV
|