Prechádzať zdrojové kódy

Add Sentencepiece (#1556)

* Add sentencepiece

* Fix checksum

* Apply fix from code review

* Update xmake.lua

* Update xmake.lua

Co-authored-by: ruki <[email protected]>
A2va 2 rokov pred
rodič
commit
159592ee13
1 zmenil súbory, kde vykonal 49 pridanie a 0 odobranie
  1. 49 0
      packages/s/sentencepiece/xmake.lua

+ 49 - 0
packages/s/sentencepiece/xmake.lua

@@ -0,0 +1,49 @@
+package("sentencepiece")
+
+    set_homepage("https://github.com/google/sentencepiece")
+    set_description("Unsupervised text tokenizer for Neural Network-based text generation. .")
+    set_license("Apache-2.0")
+
+    add_urls("https://github.com/google/sentencepiece/archive/$(version).tar.gz",
+             "https://github.com/google/sentencepiece.git")
+
+    add_versions("v0.1.97", "41c3a07f315e3ac87605460c8bb8d739955bc8e7f478caec4017ef9b7d78669b")
+
+    add_deps("cmake", "gperftools")
+
+    add_configs("external_abseil",  {description = "Use external abseil.", default = false, type = "boolean"})
+    add_configs("builtin_protobuf", {description = "Use built-in protobuf.", default = true, type = "boolean"})
+    if is_plat("windows") then
+        add_configs("shared",     {description = "Build shared library.", default = false, type = "boolean", readonly = true})
+        add_configs("vs_runtime", {description = "Set vs compiler runtime.", default = "MT", readonly = true})
+    end
+
+    on_load("windows", "linux", "macosx",function (package)
+        if package:config("external_abseil") then
+            package:add("deps", "abseil")
+        end
+        if not package:config("builtin_protobuf") then
+            package:add("deps", "protobuf-cpp")
+        end
+    end)
+
+    on_install("windows", "linux", "macosx", function (package)
+        local configs = {}
+        table.insert(configs, "-DSPM_ENABLE_SHARED=" .. (package:config("shared") and "ON" or "OFF"))
+        table.insert(configs, "-DSPM_USE_EXTERNAL_ABSL=" .. (package:config("external_abseil") and "ON" or "OFF"))
+        table.insert(configs, "-DSPM_USE_BUILTIN_PROTOBUF=" .. (package:config("builtin_protobuf") and "ON" or "OFF"))
+        import("package.tools.cmake").install(package, configs)
+    end)
+
+    on_test(function (package)
+        assert(package:check_cxxsnippets({test = [[
+            #include <iostream>
+            void test(int args, char** argv) {
+                sentencepiece::SentencePieceProcessor processor;
+                const auto status = processor.Load("//path/to/model.model");
+                if (!status.ok()) {
+                    std::cerr << status.ToString() << std::endl;
+                }
+            }
+        ]]}, {configs = {languages = "c++17"}, includes = "sentencepiece_processor.h"}))
+    end)