Fixing bug with non-default cpu memory types (#5)

Co-authored-by: Gilbert Lee <gilbertlee-amd@github.com>

CHANGELOG.md

 # Changelog for TransferBench
 
+## v1.07
+### Changed
+- Fix bug with allocations involving non-default CPU memory types
+
 ## v1.06
 ### Added
 - Added unpinned CPU memory type ('U').  May require HSA_XNACK=1 in order to access via GPU executors

EnvVars.hpp

@@ -26,7 +26,7 @@ THE SOFTWARE.
 #include <algorithm>
 #include <random>
 #include <time.h>
-#define TB_VERSION "1.06"
+#define TB_VERSION "1.07"
 
 extern char const MemTypeStr[];
 

TransferBench.cpp

@@ -546,7 +546,7 @@ int RemappedIndex(int const origIdx, MemType const memType)
   static std::vector<int> remapping;
 
   // No need to re-map CPU devices
-  if (memType == MEM_CPU) return origIdx;
+  if (IsCpuType(memType)) return origIdx;
 
   // Build remapping on first use
   if (remapping.empty())