AMDGPU: Fix handling of infinite loops in fragment shaders
authorConnor Abbott <cwabbott0@gmail.com>
Wed, 27 Nov 2019 13:09:13 +0000 (14:09 +0100)
committerConnor Abbott <cwabbott0@gmail.com>
Wed, 29 Jan 2020 14:08:46 +0000 (15:08 +0100)
commit0994c485e61322a04e580d83617eab547292aba2
tree4ae114ac078f139d44ae7f31045591f621a401e8
parentfce8983a3c03b41c3ba4bdaef72e64e29ff9ecc0
AMDGPU: Fix handling of infinite loops in fragment shaders

Summary:
Due to the fact that kill is just a normal intrinsic, even though it's
supposed to terminate the thread, we can end up with provably infinite
loops that are actually supposed to end successfully. The
AMDGPUUnifyDivergentExitNodes pass breaks up these loops, but because
there's no obvious place to make the loop branch to, it just makes it
return immediately, which skips the exports that are supposed to happen
at the end and hangs the GPU if all the threads end up being killed.

While it would be nice if the fact that kill terminates the thread were
modeled in the IR, I think that the structurizer as-is would make a mess if we
did that when the kill is inside control flow. For now, we just add a null
export at the end to make sure that it always exports something, which fixes
the immediate problem without penalizing the more common case. This means that
we sometimes do two "done" exports when only some of the threads enter the
discard loop, but from tests the hardware seems ok with that.

This fixes dEQP-VK.graphicsfuzz.while-inside-switch with radv.

Reviewers: arsenm, nhaehnle

Subscribers: kzhuravl, jvesely, wdng, yaxunl, dstuttard, tpr, t-tye, hiraditya, llvm-commits

Tags: #llvm

Differential Revision: https://reviews.llvm.org/D70781
llvm/lib/Target/AMDGPU/AMDGPUUnifyDivergentExitNodes.cpp
llvm/test/CodeGen/AMDGPU/kill-infinite-loop.ll [new file with mode: 0644]