[PATCH umr] Capture COMPUTE shaders in ring decoding as well.

[Tom St Denis]

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

diff --git a/src/lib/ring_decode.c b/src/lib/ring_decode.c
index 42265e0a74c9..7e3218a46145 100644
--- a/src/lib/ring_decode.c
+++ b/src/lib/ring_decode.c
@@ -774,9 +774,9 @@ static void print_decode_pm4_pkt3(struct umr_asic *asic, struct umr_ring_decoder
 					{
 						char *tmp = umr_reg_name(asic, decoder->pm4.next_write_mem.addr_lo);
 						printf("%s <= %s0x%08lx%s", tmp, YELLOW, (unsigned long)ib, RST);
-						if (strstr(tmp, "SPI_SHADER_PGM_LO_")) {
+						if (strstr(tmp, "SPI_SHADER_PGM_LO_") || strstr(tmp, "COMPUTE_PGM_LO")) {
 							decoder->pm4.next_ib_state.ib_addr_lo = ib;
-						} else if (strstr(tmp, "SPI_SHADER_PGM_HI_")) {
+						} else if (strstr(tmp, "SPI_SHADER_PGM_HI_") || strstr(tmp, "COMPUTE_PGM_HI")) {
 							decoder->pm4.next_ib_state.ib_addr_hi = ib;
 							decoder->pm4.next_ib_state.ib_vmid = decoder->next_ib_info.vmid;
 							add_shader(asic, decoder);
-- 
2.14.3