Add rccl API support (#66)

* [Draft]: Add rccl API support

* Partial tests

Need to add tests to the cmake file
This commit is contained in:
Mythreya
2024-09-11 15:57:51 -07:00
committed by GitHub
parent 11a4668306
commit b71f9cabe6
8 changed files with 267 additions and 0 deletions
@@ -105,6 +105,7 @@ enum rocp_reg_supported_library // NOLINT(performance-enum-size)
ROCP_REG_HIP,
ROCP_REG_ROCTX,
ROCP_REG_HIP_COMPILER,
ROCP_REG_RCCL,
ROCP_REG_LAST,
};
@@ -159,6 +160,11 @@ ROCP_REG_DEFINE_LIBRARY_TRAITS(ROCP_REG_HIP_COMPILER,
"rocprofiler_register_import_hip_compiler",
"libamdhip64.so.[6-9]($|\\.[0-9\\.]+)")
ROCP_REG_DEFINE_LIBRARY_TRAITS(ROCP_REG_RCCL,
"rccl",
"rocprofiler_register_import_rccl",
"librccl.so.[6-9]($|\\.[0-9\\.]+)")
ROCP_REG_DEFINE_ERROR_MESSAGE(ROCP_REG_SUCCESS, "Success")
ROCP_REG_DEFINE_ERROR_MESSAGE(ROCP_REG_NO_TOOLS, "rocprofiler-register found no tools")
ROCP_REG_DEFINE_ERROR_MESSAGE(ROCP_REG_DEADLOCK, "rocprofiler-register deadlocked")