added include directo

auto-ndp · Feb 14, 2024 · eaeb790 · eaeb790
1 parent f6cef20
commit eaeb790
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -106,6 +106,7 @@ add_subdirectory(src/mpi)
 add_subdirectory(src/proto)
 add_subdirectory(src/redis)
 add_subdirectory(src/runner)
+add_subdirectory(src/loadbalance)
 add_subdirectory(src/scheduler)
 add_subdirectory(src/snapshot)
 add_subdirectory(src/state)
@@ -130,6 +131,7 @@ add_library(faabric
         $<TARGET_OBJECTS:scheduler_obj>
         $<TARGET_OBJECTS:snapshot_obj>
         $<TARGET_OBJECTS:state_obj>
+        $<TARGET_OBJECTS:loadbalance_obj>
         $<TARGET_OBJECTS:transport_obj>
         $<TARGET_OBJECTS:util_obj>
         )

diff --git a/src/scheduler/Scheduler.cpp b/src/scheduler/Scheduler.cpp
@@ -20,6 +20,7 @@
 #include <faabric/util/string_tools.h>
 #include <faabric/util/testing.h>
 #include <faabric/util/timing.h>
+#include <faabric/loadbalance/LoadBalancePolicy.h>
 
 #include <sys/eventfd.h>
 #include <sys/file.h>