Add meta info function for AMD ROCm support

breiters · Nov 9, 2023 · 141a59e · 141a59e
1 parent 3b63ad7
commit 141a59e
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/src/includes/likwid.h b/src/includes/likwid.h
@@ -83,9 +83,13 @@ extern int likwid_getMinorVersion(void) __attribute__ ((visibility ("default") )
 @return Bugfix version
 */
 extern int likwid_getBugfixVersion(void) __attribute__ ((visibility ("default") ));
+
 /*! \brief Check whether the LIKWID library was built with Nvidia GPU support */
 extern int likwid_getNvidiaSupport(void) __attribute__ ((visibility ("default") ));
 
+/*! \brief Check whether the LIKWID library was built with AMD GPU support */
+extern int likwid_getRocmSupport(void) __attribute__ ((visibility ("default") ));
+
 /*! \brief Get the maximal count of supported HW threads */
 extern int likwid_getMaxSupportedThreads(void) __attribute__ ((visibility ("default") ));
 

diff --git a/src/meta.c b/src/meta.c
@@ -27,6 +27,15 @@ int likwid_getNvidiaSupport(void)
 #endif
 }
 
+int likwid_getRocmSupport(void)
+{
+#ifdef LIKWID_WITH_ROCMON
+    return 1;
+#else
+    return 0;
+#endif
+}
+
 int likwid_getMaxSupportedThreads(void)
 {
     return (int) MAX_NUM_THREADS;