glsl: Use gl_SubGroupInvocationARB
This commit is contained in:
parent
e35ffbbeb0
commit
68ef3803bf
|
@ -151,7 +151,6 @@ void EmitContext::SetupExtensions(std::string&) {
|
||||||
info.uses_subgroup_shuffles || info.uses_fswzadd) {
|
info.uses_subgroup_shuffles || info.uses_fswzadd) {
|
||||||
header += "#extension GL_ARB_shader_ballot : enable\n";
|
header += "#extension GL_ARB_shader_ballot : enable\n";
|
||||||
header += "#extension GL_ARB_shader_group_vote : enable\n";
|
header += "#extension GL_ARB_shader_group_vote : enable\n";
|
||||||
header += "#extension GL_KHR_shader_subgroup_basic : enable\n";
|
|
||||||
if (!info.uses_int64) {
|
if (!info.uses_int64) {
|
||||||
header += "#extension GL_ARB_gpu_shader_int64 : enable\n";
|
header += "#extension GL_ARB_gpu_shader_int64 : enable\n";
|
||||||
}
|
}
|
||||||
|
|
|
@ -45,8 +45,8 @@ void EmitVoteAll(EmitContext& ctx, IR::Inst& inst, std::string_view pred) {
|
||||||
if (!ctx.profile.warp_size_potentially_larger_than_guest) {
|
if (!ctx.profile.warp_size_potentially_larger_than_guest) {
|
||||||
ctx.AddU1("{}=allInvocationsEqualARB({});", inst, pred);
|
ctx.AddU1("{}=allInvocationsEqualARB({});", inst, pred);
|
||||||
} else {
|
} else {
|
||||||
const auto active_mask{fmt::format("uvec2(ballotARB(true))[gl_SubgroupInvocationID]")};
|
const auto active_mask{fmt::format("uvec2(ballotARB(true))[gl_SubGroupInvocationARB]")};
|
||||||
const auto ballot{fmt::format("uvec2(ballotARB({}))[gl_SubgroupInvocationID]", pred)};
|
const auto ballot{fmt::format("uvec2(ballotARB({}))[gl_SubGroupInvocationARB]", pred)};
|
||||||
ctx.AddU1("{}=({}&{})=={};", inst, ballot, active_mask, active_mask);
|
ctx.AddU1("{}=({}&{})=={};", inst, ballot, active_mask, active_mask);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -55,8 +55,8 @@ void EmitVoteAny(EmitContext& ctx, IR::Inst& inst, std::string_view pred) {
|
||||||
if (!ctx.profile.warp_size_potentially_larger_than_guest) {
|
if (!ctx.profile.warp_size_potentially_larger_than_guest) {
|
||||||
ctx.AddU1("{}=anyInvocationARB({});", inst, pred);
|
ctx.AddU1("{}=anyInvocationARB({});", inst, pred);
|
||||||
} else {
|
} else {
|
||||||
const auto active_mask{fmt::format("uvec2(ballotARB(true))[gl_SubgroupInvocationID]")};
|
const auto active_mask{fmt::format("uvec2(ballotARB(true))[gl_SubGroupInvocationARB]")};
|
||||||
const auto ballot{fmt::format("uvec2(ballotARB({}))[gl_SubgroupInvocationID]", pred)};
|
const auto ballot{fmt::format("uvec2(ballotARB({}))[gl_SubGroupInvocationARB]", pred)};
|
||||||
ctx.AddU1("{}=({}&{})!=0u;", inst, ballot, active_mask, active_mask);
|
ctx.AddU1("{}=({}&{})!=0u;", inst, ballot, active_mask, active_mask);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -65,8 +65,8 @@ void EmitVoteEqual(EmitContext& ctx, IR::Inst& inst, std::string_view pred) {
|
||||||
if (!ctx.profile.warp_size_potentially_larger_than_guest) {
|
if (!ctx.profile.warp_size_potentially_larger_than_guest) {
|
||||||
ctx.AddU1("{}=allInvocationsEqualARB({});", inst, pred);
|
ctx.AddU1("{}=allInvocationsEqualARB({});", inst, pred);
|
||||||
} else {
|
} else {
|
||||||
const auto active_mask{fmt::format("uvec2(ballotARB(true))[gl_SubgroupInvocationID]")};
|
const auto active_mask{fmt::format("uvec2(ballotARB(true))[gl_SubGroupInvocationARB]")};
|
||||||
const auto ballot{fmt::format("uvec2(ballotARB({}))[gl_SubgroupInvocationID]", pred)};
|
const auto ballot{fmt::format("uvec2(ballotARB({}))[gl_SubGroupInvocationARB]", pred)};
|
||||||
const auto value{fmt::format("({}^{})", ballot, active_mask)};
|
const auto value{fmt::format("({}^{})", ballot, active_mask)};
|
||||||
ctx.AddU1("{}=({}==0)||({}=={});", inst, value, value, active_mask);
|
ctx.AddU1("{}=({}==0)||({}=={});", inst, value, value, active_mask);
|
||||||
}
|
}
|
||||||
|
@ -76,7 +76,7 @@ void EmitSubgroupBallot(EmitContext& ctx, IR::Inst& inst, std::string_view pred)
|
||||||
if (!ctx.profile.warp_size_potentially_larger_than_guest) {
|
if (!ctx.profile.warp_size_potentially_larger_than_guest) {
|
||||||
ctx.AddU32("{}=uvec2(ballotARB({})).x;", inst, pred);
|
ctx.AddU32("{}=uvec2(ballotARB({})).x;", inst, pred);
|
||||||
} else {
|
} else {
|
||||||
ctx.AddU32("{}=uvec2(ballotARB({}))[gl_SubgroupInvocationID];", inst, pred);
|
ctx.AddU32("{}=uvec2(ballotARB({}))[gl_SubGroupInvocationARB];", inst, pred);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Reference in New Issue