|
@@ -0,0 +1,99 @@
|
|
|
|
+package("ggml")
|
|
|
|
+ set_homepage("https://github.com/ggml-org/ggml")
|
|
|
|
+ set_description("Tensor library for machine learning")
|
|
|
|
+ set_license("MIT")
|
|
|
|
+
|
|
|
|
+ add_urls("https://github.com/ggml-org/ggml.git", {submodules = false})
|
|
|
|
+
|
|
|
|
+ add_versions("2025.03.05", "d013aa56cdcccefd9086ac93a58951256c84ff16")
|
|
|
|
+
|
|
|
|
+ add_configs("openmp", {description = "use OpenMP", default = false, type = "boolean"})
|
|
|
|
+ add_configs("cuda", {description = "use CUDA", default = false, type = "boolean"})
|
|
|
|
+ add_configs("vulkan", {description = "use Vulkan", default = false, type = "boolean"})
|
|
|
|
+ add_configs("kcompute", {description = "use Kompute", default = false, type = "boolean"})
|
|
|
|
+ add_configs("blas", {description = "use BLAS", default = nil, type = "string", values = {"mkl", "openblas"}})
|
|
|
|
+
|
|
|
|
+ add_deps("cmake")
|
|
|
|
+
|
|
|
|
+ on_check("windows", function (package)
|
|
|
|
+ if package:is_arch("arm.*") and package:has_tool("cxx", "cl") then
|
|
|
|
+ raise("package(ggml) MSVC is not supported for ARM, use clang")
|
|
|
|
+ end
|
|
|
|
+ end)
|
|
|
|
+
|
|
|
|
+ on_load(function (package)
|
|
|
|
+ if package:config("openmp") then
|
|
|
|
+ package:add("deps", "openmp")
|
|
|
|
+ end
|
|
|
|
+ if package:config("cuda") then
|
|
|
|
+ package:add("deps", "cuda")
|
|
|
|
+ end
|
|
|
|
+ if package:config("kcompute") then
|
|
|
|
+ package:add("deps", "kcompute")
|
|
|
|
+ end
|
|
|
|
+ if package:config("vulkan") then
|
|
|
|
+ -- requires vulkan loader and glslc
|
|
|
|
+ package:add("deps", "vulkansdk")
|
|
|
|
+ package:add("deps", "shaderc", {configs = {binaryonly = true}})
|
|
|
|
+ end
|
|
|
|
+ if package:config("blas") then
|
|
|
|
+ -- TODO: test it
|
|
|
|
+ if is_subhost("windows") then
|
|
|
|
+ package:add("deps", "pkgconf")
|
|
|
|
+ else
|
|
|
|
+ package:add("deps", "pkg-config")
|
|
|
|
+ end
|
|
|
|
+ end
|
|
|
|
+
|
|
|
|
+ if package:config("shared") then
|
|
|
|
+ package:add("defines", "GGML_SHARED")
|
|
|
|
+ end
|
|
|
|
+
|
|
|
|
+ local backends = {
|
|
|
|
+ "ggml-cpu",
|
|
|
|
+ "ggml-cuda",
|
|
|
|
+ "ggml-vulkan",
|
|
|
|
+ }
|
|
|
|
+ package:add("links", "ggml", backends, "ggml-base")
|
|
|
|
+
|
|
|
|
+ if package:is_plat("linux", "bsd") then
|
|
|
|
+ package:add("syslinks", "pthread", "dl", "m")
|
|
|
|
+ elseif package:is_plat("android") then
|
|
|
|
+ package:add("syslinks", "dl")
|
|
|
|
+ end
|
|
|
|
+ end)
|
|
|
|
+
|
|
|
|
+ on_install("!cross", function (package)
|
|
|
|
+ -- Fix missing prefix `lib` for mingw
|
|
|
|
+ io.replace("CMakeLists.txt", "# remove the lib prefix on win32 mingw\nif (WIN32)", "if(0)", {plain = true})
|
|
|
|
+
|
|
|
|
+ local configs = {
|
|
|
|
+ "-DGGML_ALL_WARNINGS=OFF",
|
|
|
|
+ "-DGGML_BUILD_TESTS=OFF",
|
|
|
|
+ "-DGGML_BUILD_EXAMPLES=OFF",
|
|
|
|
+ "-DGGML_CCACHE=OFF",
|
|
|
|
+ }
|
|
|
|
+ table.insert(configs, "-DCMAKE_BUILD_TYPE=" .. (package:is_debug() and "Debug" or "Release"))
|
|
|
|
+ table.insert(configs, "-DBUILD_SHARED_LIBS=" .. (package:config("shared") and "ON" or "OFF"))
|
|
|
|
+ table.insert(configs, "-DGGML_LTO=" .. (package:config("lto") and "ON" or "OFF"))
|
|
|
|
+ table.insert(configs, "-DGGML_SANITIZE_ADDRESS=" .. (package:config("asan") and "ON" or "OFF"))
|
|
|
|
+
|
|
|
|
+ table.insert(configs, "-DGGML_OPENMP=" .. (package:config("openmp") and "ON" or "OFF"))
|
|
|
|
+ table.insert(configs, "-DGGML_CUDA=" .. (package:config("cuda") and "ON" or "OFF"))
|
|
|
|
+ table.insert(configs, "-DGGML_VULKAN=" .. (package:config("vulkan") and "ON" or "OFF"))
|
|
|
|
+ table.insert(configs, "-DGGML_OPENBLAS=" .. (package:config("blas") and "ON" or "OFF"))
|
|
|
|
+ table.insert(configs, "-DGGML_KOMPUTE=" .. (package:config("kcompute") and "ON" or "OFF"))
|
|
|
|
+
|
|
|
|
+ local opt = {}
|
|
|
|
+ if package:has_tool("cxx", "cl") then
|
|
|
|
+ opt.cxflags = "/utf-8"
|
|
|
|
+ if package:config("cuda") then
|
|
|
|
+ table.insert(configs, "-DCMAKE_CUDA_FLAGS=-Xcompiler /utf-8")
|
|
|
|
+ end
|
|
|
|
+ end
|
|
|
|
+ import("package.tools.cmake").install(package, configs, opt)
|
|
|
|
+ end)
|
|
|
|
+
|
|
|
|
+ on_test(function (package)
|
|
|
|
+ assert(package:has_cfuncs("ggml_time_init", {includes = "ggml.h"}))
|
|
|
|
+ end)
|