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

!39376 Add FastTransformer to third party package 

Merge pull request !39376 from yonibaehr/export_yoni
This commit is contained in:
i-robot 2022-08-11 02:54:10 +00:00 committed by Gitee
parent fac6cab21b aac8073a2e
commit 9fb86b5edb
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
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)
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()

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