work stealing chunked task allocator example for issue #291

ggerganov · Jun 27, 2023 · b1d402d · b1d402d
1 parent 0be54f7
commit b1d402d
Show file tree

Hide file tree

Showing 3 changed files with 441 additions and 0 deletions.
diff --git a/examples/CMakeLists.txt b/examples/CMakeLists.txt
@@ -39,6 +39,7 @@ else()
     add_subdirectory(baby-llama)
     add_subdirectory(train-text-from-scratch)
     add_subdirectory(simple)
+    add_subdirectory(task-allocator)
     if (LLAMA_METAL)
         add_subdirectory(metal)
     endif()

diff --git a/examples/task-allocator/CMakeLists.txt b/examples/task-allocator/CMakeLists.txt
@@ -0,0 +1,7 @@
+set(TARGET task-allocator)
+add_executable(${TARGET} task-allocator.c)
+target_link_libraries(${TARGET} PRIVATE)
+target_compile_features(${TARGET} PRIVATE c_std_11)
+if(TARGET BUILD_INFO)
+  add_dependencies(${TARGET} BUILD_INFO)
+endif()