fix(docker): build CPU dockers

jolibrain · Sep 5, 2023 · 9e56aba · 9e56aba
1 parent e1118b1
commit 9e56aba
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -446,6 +446,11 @@ if (CUDA_FOUND)
   message(STATUS "CUDA_ARCH=${CUDA_ARCH}")
 else()
   set(CUDA_LIB_DEPS "")
+
+  if (NOT APPLE)
+    # XXX: On Apple platforms MPS can be used, so we'd need a different condition
+    add_definitions(-DCPU_ONLY)
+  endif()
 endif()
 
 if (USE_DLIB)