From ebfc82a692c1fd7a0f492afe3961bc5886a830f4 Mon Sep 17 00:00:00 2001 From: rohithkrn Date: Fri, 22 Sep 2023 14:34:25 -0700 Subject: [PATCH] build ft for sm90 --- tools/scripts/build_ft_deps.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/scripts/build_ft_deps.sh b/tools/scripts/build_ft_deps.sh index 4d3cb94a103..110bedc2d92 100755 --- a/tools/scripts/build_ft_deps.sh +++ b/tools/scripts/build_ft_deps.sh @@ -41,7 +41,7 @@ cd ../../ mkdir -p FasterTransformer/build cd FasterTransformer/build git submodule init && git submodule update -cmake -DCMAKE_BUILD_TYPE=Release -DSM=70,75,80,86 -DBUILD_PYT=ON -DBUILD_MULTI_GPU=ON .. +cmake -DCMAKE_BUILD_TYPE=Release -DSM=70,75,80,86,90 -DBUILD_PYT=ON -DBUILD_MULTI_GPU=ON .. make -j$(nproc) cp lib/libth_transformer.so /tmp/binaries/ cd ../../