Move GPU devices management into hardware subfolder

[alexxy/gromacs.git] / src / gromacs / hardware / CMakeLists.txt

diff --git a/src/gromacs/hardware/CMakeLists.txt b/src/gromacs/hardware/CMakeLists.txt
index efc684c31f34df105fe71bbc5e365ddab6d4b7f6..c93767f790ee008717e341c6b0a9bb0023e641b7 100644 (file)
--- a/src/gromacs/hardware/CMakeLists.txt
+++ b/src/gromacs/hardware/CMakeLists.txt
@@ -35,11 +35,26 @@
 gmx_add_libgromacs_sources(
     cpuinfo.cpp
     detecthardware.cpp
+    device_management_common.cpp
     hardwaretopology.cpp
     printhardware.cpp
     identifyavx512fmaunits.cpp
     )
 
+if(GMX_GPU_OPENCL)
+    gmx_add_libgromacs_sources(
+        device_management_ocl.cpp
+        )
+elseif(GMX_GPU_CUDA)
+    gmx_add_libgromacs_sources(
+        device_management.cu
+        )
+else()
+    gmx_add_libgromacs_sources(
+        device_management.cpp
+    )
+endif()
+
 if (BUILD_TESTING)
     add_subdirectory(tests)
 endif()