From c64b5e0090bcab54351e6d599013348dd061b3e5 Mon Sep 17 00:00:00 2001
From: Giovanni  LB <gbaraldi@amd.com>
Date: Sat, 15 Apr 2023 19:34:18 -0300
Subject: [PATCH] PLAT-130888: Added back shader engine mask.

Change-Id: I8e14bbd16628cc0b682adea845911b93b8e1af51
---
 src/tools/tool.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/tools/tool.cpp b/src/tools/tool.cpp
index 7b2f857f5a..395ce7432c 100644
--- a/src/tools/tool.cpp
+++ b/src/tools/tool.cpp
@@ -187,6 +187,7 @@ att_parsed_input_t GetATTParams() {
   std::unordered_map<std::string, rocprofiler_att_parameter_name_t> ATT_PARAM_NAMES{};
 
   ATT_PARAM_NAMES["att: TARGET_CU"] = ROCPROFILER_ATT_COMPUTE_UNIT_TARGET;
+  ATT_PARAM_NAMES["SE_MASK"] = ROCPROFILER_ATT_SE_MASK;
   ATT_PARAM_NAMES["SIMD_MASK"] = ROCPROFILER_ATT_MAXVALUE;
   ATT_PARAM_NAMES["PERFCOUNTER_ID"] = ROCPROFILER_ATT_PERFCOUNTER;
   ATT_PARAM_NAMES["PERFCOUNTER"] = ROCPROFILER_ATT_PERFCOUNTER_NAME;
@@ -198,7 +199,7 @@ att_parsed_input_t GetATTParams() {
   std::unordered_map<std::string, uint32_t> default_params = {
     {"ATT_MASK", 0x3F01},
     {"TOKEN_MASK", 0x344B},
-    {"TOKEN_MASK2", 0x3FFFF}
+    {"TOKEN_MASK2", 0xFFFFFFF}
   };
 
   bool started_att_counters = false;