Add Fast transformers to mindspore third party

This commit is contained in:
yoni 2022-08-08 12:18:58 +03:00
parent db23bd8b8d
commit aac8073a2e
3 changed files with 3186 additions and 0 deletions

View File

@ -0,0 +1,13 @@
set(REQ_URL "https://github.com/NVIDIA/FasterTransformer/archive/refs/tags/release/v5.0_tag.tar.gz")
set(MD5 "f2e06ec43f3b5b83017bd87b0427524f")
set(ft_libs "transformer-shared")
mindspore_add_pkg(fast_transformers
VER 0.5.0
URL ${REQ_URL}
MD5 ${MD5}
LIBS ${ft_libs}
LIB_PATH output/lib
PATCHES ${MINDSPORE_PROJECT_DIR}/third_party/patch/fast_transformer/001-fast_transformer.patch
CMAKE_OPTION -DCMAKE_BUILD_TYPE=Release -DEXAMPLES=off)

View File

@ -49,6 +49,7 @@ endif()
if(ENABLE_GPU)
include(${CMAKE_SOURCE_DIR}/cmake/external_libs/cub.cmake)
include(${CMAKE_SOURCE_DIR}/cmake/external_libs/fast_transformers.cmake)
if(ENABLE_MPI)
include(${CMAKE_SOURCE_DIR}/cmake/external_libs/nccl.cmake)
endif()

File diff suppressed because it is too large Load Diff