diff --git a/packages/f/fbgemm/patches/0.7.0/dep-unbundle.patch b/packages/f/fbgemm/patches/0.7.0/dep-unbundle.patch new file mode 100644 index 000000000..078616e5c --- /dev/null +++ b/packages/f/fbgemm/patches/0.7.0/dep-unbundle.patch @@ -0,0 +1,57 @@ +diff --git a/CMakeLists.txt b/CMakeLists.txt +index 0b51071..ecc2c9e 100644 +--- a/CMakeLists.txt ++++ b/CMakeLists.txt +@@ -240,8 +240,8 @@ message(WARNING "CMAKE_BUILD_TYPE = ${CMAKE_BUILD_TYPE}") + message(WARNING "CMAKE_CXX_FLAGS_DEBUG is ${CMAKE_CXX_FLAGS_DEBUG}") + message(WARNING "CMAKE_CXX_FLAGS_RELEASE is ${CMAKE_CXX_FLAGS_RELEASE}") + message(WARNING "==========") +- +-if(NOT TARGET asmjit) ++find_package(asmjit CONFIG REQUIRED) ++if(0) + #Download asmjit from github if ASMJIT_SRC_DIR is not specified. + if(NOT DEFINED ASMJIT_SRC_DIR) + set(ASMJIT_SRC_DIR "${FBGEMM_SOURCE_DIR}/third_party/asmjit" +@@ -269,8 +269,8 @@ if(NOT TARGET asmjit) + target_compile_options_if_supported(asmjit -Wno-unused-but-set-variable) + endif() + endif() +- +-if(NOT TARGET cpuinfo) ++find_package(cpuinfo CONFIG REQUIRED) ++if(0) + #Download cpuinfo from github if CPUINFO_SOURCE_DIR is not specified. + if(NOT DEFINED CPUINFO_SOURCE_DIR) + set(CPUINFO_SOURCE_DIR "${FBGEMM_SOURCE_DIR}/third_party/cpuinfo" +@@ -354,12 +354,7 @@ target_include_directories(fbgemm BEFORE + PUBLIC $ + PUBLIC $) + +-target_link_libraries(fbgemm +- $ +- $) +-add_dependencies(fbgemm +- asmjit +- cpuinfo) ++target_link_libraries(fbgemm asmjit cpuinfo) + + if(OpenMP_FOUND) + target_link_libraries(fbgemm OpenMP::OpenMP_CXX) +@@ -383,12 +378,12 @@ install( + + if(MSVC) + if(FBGEMM_LIBRARY_TYPE STREQUAL "shared") +- install( +- FILES $ $ +- DESTINATION ${CMAKE_INSTALL_LIBDIR} OPTIONAL) ++ # install( ++ # FILES $ $ ++ # DESTINATION ${CMAKE_INSTALL_LIBDIR} OPTIONAL) + endif() + install(TARGETS fbgemm DESTINATION ${CMAKE_INSTALL_LIBDIR}) +- install(TARGETS asmjit DESTINATION ${CMAKE_INSTALL_LIBDIR}) ++ # install(TARGETS asmjit DESTINATION ${CMAKE_INSTALL_LIBDIR}) + endif() + + #Make project importable from the build directory diff --git a/packages/f/fbgemm/xmake.lua b/packages/f/fbgemm/xmake.lua new file mode 100644 index 000000000..a22bc9b32 --- /dev/null +++ b/packages/f/fbgemm/xmake.lua @@ -0,0 +1,46 @@ +package("fbgemm") + set_homepage("https://github.com/pytorch/FBGEMM") + set_description("FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/") + set_license("BSD") + + add_urls("https://github.com/pytorch/FBGEMM/archive/refs/tags/$(version).tar.gz", + "https://github.com/pytorch/FBGEMM.git") + + add_versions("v0.7.0", "c51ac26bc0aa8fef7e80631c4abdd3a7c33d1a097359cef9b008bf9e1203c071") + + add_patches("0.7.0", "patches/0.7.0/dep-unbundle.patch", "f3117ff728989146d5ab0c370fe410c73459091f65cae5f6b304e5637889fb8f") + + -- need libtorch + add_configs("gpu", {description = "Build fbgemm_gpu library", default = false, type = "boolean"}) + add_configs("cpu", {description = "Build FBGEMM_GPU without GPU support", default = false, type = "boolean"}) + add_configs("rocm", {description = "Build FBGEMM_GPU for ROCm", default = false, type = "boolean"}) + + add_deps("cmake", "python", {kind = "binary"}) + add_deps("asmjit", "cpuinfo", "openmp") + + -- mingw support: https://github.com/pytorch/FBGEMM/pull/2114 + -- arm support: https://github.com/pytorch/FBGEMM/issues/2074 + on_install("windows|x64", "linux|x86_64", "macosx|x86_64", function (package) + io.replace("CMakeLists.txt", "-Werror", "", {plain = true}) + + if not package:config("shared") then + package:add("defines", "FBGEMM_STATIC") + end + + local configs = {"-DFBGEMM_BUILD_TESTS=OFF", "-DFBGEMM_BUILD_BENCHMARKS=OFF"} + table.insert(configs, "-DCMAKE_BUILD_TYPE=" .. (package:is_debug() and "Debug" or "Release")) + table.insert(configs, "-DFBGEMM_LIBRARY_TYPE=" .. (package:config("shared") and "shared" or "static")) + + table.insert(configs, "-DFBGEMM_BUILD_FBGEMM_GPU=" .. (package:config("gpu") and "ON" or "OFF")) + table.insert(configs, "-DFBGEMM_CPU_ONLY=" .. (package:config("cpu") and "ON" or "OFF")) + table.insert(configs, "-DUSE_ROCM=" .. (package:config("rocm") and "ON" or "OFF")) + import("package.tools.cmake").install(package, configs, {packagedeps = {"asmjit", "cpuinfo"}}) + end) + + on_test(function (package) + assert(package:check_cxxsnippets({test = [[ + void test() { + fbgemm::Xor128(); + } + ]]}, {configs = {languages = "c++20"}, includes = "fbgemm/QuantUtilsAvx2.h"})) + end)