Mesa (vulkan): nir/lower_io: Add UBOs and SSBOs to get_io_offset_src

[Jason Ekstrand]

Module: Mesa
Branch: vulkan
Commit: d7cddbd6d66120899ffcb5735ee436f13432ed64
URL:    http://cgit.freedesktop.org/mesa/mesa/commit/?id=d7cddbd6d66120899ffcb5735ee436f13432ed64

Author: Jason Ekstrand <jason.ekstrand at intel.com>
Date:   Thu Apr 14 10:31:27 2016 -0700

nir/lower_io: Add UBOs and SSBOs to get_io_offset_src

Reviewed-by: Kenneth Graunke <kenneth at whitecape.org>

---

 src/compiler/nir/nir_lower_io.c | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/compiler/nir/nir_lower_io.c b/src/compiler/nir/nir_lower_io.c
index 6d4a3d8..df1f7a5 100644
--- a/src/compiler/nir/nir_lower_io.c
+++ b/src/compiler/nir/nir_lower_io.c
@@ -430,10 +430,13 @@ nir_get_io_offset_src(nir_intrinsic_instr *instr)
    case nir_intrinsic_load_output:
    case nir_intrinsic_load_uniform:
       return &instr->src[0];
+   case nir_intrinsic_load_ubo:
+   case nir_intrinsic_load_ssbo:
    case nir_intrinsic_load_per_vertex_input:
    case nir_intrinsic_load_per_vertex_output:
    case nir_intrinsic_store_output:
       return &instr->src[1];
+   case nir_intrinsic_store_ssbo:
    case nir_intrinsic_store_per_vertex_output:
       return &instr->src[2];
    default: