From 24fcf64ed19bb178aa867b14c1d7f13493656e74 Mon Sep 17 00:00:00 2001
From: David Addison <daddison@nvidia.com>
Date: Tue, 22 Nov 2022 11:18:37 -0800
Subject: [PATCH] Call cudaFreeHost() on wrongPerGpu not cudaFree()

---
 src/common.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/common.cu b/src/common.cu
index 41d7479..48a629c 100644
--- a/src/common.cu
+++ b/src/common.cu
@@ -262,7 +262,7 @@ testResult_t CheckData(struct threadArgs* args, ncclDataType_t type, ncclRedOp_t
 
   *wrongElts = 0;
   for (int i=0; i < args->nGpus; i++) *wrongElts += wrongPerGpu[i];
-  cudaFree(wrongPerGpu);
+  cudaFreeHost(wrongPerGpu);
 
   if (args->reportErrors && *wrongElts) args->errors[0]++;
   return testSuccess;