radeonsi/gfx9: get GS wave ID from the correct input