diff --git a/tools/perf/ucc_pt_config.cc b/tools/perf/ucc_pt_config.cc index 409ef2be94..3fcb2b01c9 100644 --- a/tools/perf/ucc_pt_config.cc +++ b/tools/perf/ucc_pt_config.cc @@ -60,6 +60,7 @@ const std::map ucc_pt_memtype_map = { {"host", UCC_MEMORY_TYPE_HOST}, {"cuda", UCC_MEMORY_TYPE_CUDA}, {"rocm", UCC_MEMORY_TYPE_ROCM}, + {"cuda-mng", UCC_MEMORY_TYPE_CUDA_MANAGED}, }; const std::map ucc_pt_datatype_map = {