build ft for sm90 (#2785)

deepjavalibrary · Sep 25, 2023 · 15fd0d0 · 15fd0d0
1 parent da15713
commit 15fd0d0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 ../../