ROCm: crude attempt at detecting when ROCm is installed (issue #99)

hank-ai · Dec 31, 2024 · 5032f01 · 5032f01
1 parent 0431e6f
commit 5032f01
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/CM_dependencies.cmake b/CM_dependencies.cmake
@@ -81,7 +81,7 @@ ENDIF ()
 # =============
 # == AMD GPU ==
 # =============
-IF (NOT DARKNET_USE_CUDA) # TODO: how to detect AMD ROCm from within CMake?
+IF (NOT DARKNET_USE_CUDA AND EXISTS "/opt/rocm/include/rocm-core/rocm_version.h") # TODO: how to detect AMD ROCm from within CMake?
 	MESSAGE (STATUS "AMD ROCm detected. Darknet will use AMD GPUs.")
 	SET (DARKNET_LINK_LIBS ${DARKNET_LINK_LIBS} /opt/rocm/lib/librocm-core.so /opt/rocm/lib/librocm_smi64.so)
 	INCLUDE_DIRECTORIES ("/opt/rocm/include/")