From 15fd0d0e061b1902c26acd596f2165e7bb9a8b27 Mon Sep 17 00:00:00 2001 From: rohithkrn Date: Mon, 25 Sep 2023 10:15:27 -0700 Subject: [PATCH] build ft for sm90 (#2785) --- 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 ../../