[PATCH umr] Switch GRBM index when reading wave data directly.

[Tom St Denis]

Signed-off-by: Tom St Denis <tom.stdenis at amd.com>
---
 src/lib/wave_status.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/lib/wave_status.c b/src/lib/wave_status.c
index fe2add779fdd..7f0130bb9347 100644
--- a/src/lib/wave_status.c
+++ b/src/lib/wave_status.c
@@ -116,7 +116,9 @@ static int umr_get_wave_status_vi(struct umr_asic *asic, unsigned se, unsigned s
 		read(asic->fd.wave, &buf, 32*4);
 	} else {
 		int n = 0;
+		umr_grbm_select_index(asic, se, sh, cu);
 		read_wave_status_via_mmio(asic, simd, wave, &buf[0], &n);
+		umr_grbm_select_index(asic, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
 	}
 
 	if (buf[0] != 0) {
@@ -218,7 +220,9 @@ static int umr_get_wave_status_ai(struct umr_asic *asic, unsigned se, unsigned s
 		read(asic->fd.wave, &buf, 32*4);
 	} else {
 		int n = 0;
+		umr_grbm_select_index(asic, se, sh, cu);
 		read_wave_status_via_mmio(asic, simd, wave, &buf[0], &n);
+		umr_grbm_select_index(asic, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
 	}
 
 	if (buf[0] != 1) {
-- 
2.12.0