p27049618
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
cmake/external_libs/fast_transformers.cmake Normal file
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)

1
cmake/mind_expression.cmake
View File

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

3172
third_party/patch/fast_transformer/001-fast_transformer.patch vendored Normal file
View File

File diff suppressed because it is too large Load Diff