forked from mindspore-Ecosystem/mindspore
Add Fast transformers to mindspore third party
This commit is contained in:
parent
db23bd8b8d
commit
aac8073a2e
|
@ -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)
|
|
@ -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
Loading…
Reference in New Issue