[PATCH i-g-t] tests/intel/xe_pmu: Ensure stash_gt_freq allocates sufficient storage

[Matt Roper <matthew.d.roper@intel.com>]

stash_gt_freq() malloc's arrays based on the number of GTs returned by
the kernel's query ioctl and then indexes into the arrays by uapi GT ID.
This works fine if all GT IDs are consecutive (which is always the case
on platforms today), but will be insufficient if future platforms allow
the list to contain non-consecutive IDs (e.g., 0, 2, ...).  Allocate
storage based on the maximum GT ID returned, not the number of GTs
returned.  Some of the array indices will remain unused, but the
xe_for_each_gt() loops will already skip over those unused entries
properly.

Signed-off-by: Matt Roper <matthew.d.roper@intel.com>
---
 tests/intel/xe_pmu.c | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/tests/intel/xe_pmu.c b/tests/intel/xe_pmu.c
index a8ceac46c..ea8950ee5 100644
--- a/tests/intel/xe_pmu.c
+++ b/tests/intel/xe_pmu.c
@@ -1039,12 +1039,12 @@ static void unprovision_and_disable_vfs(int fd)
 
 static void stash_gt_freq(int fd, uint32_t **stash_min, uint32_t **stash_max)
 {
-	int num_gts, gt;
+	int max_gt, gt;
 
-	num_gts = xe_number_gt(fd);
+	max_gt = xe_dev_max_gt(fd);
 
-	*stash_min = (uint32_t *) malloc(sizeof(uint32_t) * num_gts);
-	*stash_max = (uint32_t *) malloc(sizeof(uint32_t) * num_gts);
+	*stash_min = (uint32_t *)calloc(max_gt + 1, sizeof(uint32_t));
+	*stash_max = (uint32_t *)calloc(max_gt + 1, sizeof(uint32_t));
 
 	igt_skip_on(*stash_min == NULL || *stash_max == NULL);
 
-- 
2.55.0