Skip to content

Commit 15fd0d0

Browse files
authored
build ft for sm90 (#2785)
1 parent da15713 commit 15fd0d0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tools/scripts/build_ft_deps.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ cd ../../
4141
mkdir -p FasterTransformer/build
4242
cd FasterTransformer/build
4343
git submodule init && git submodule update
44-
cmake -DCMAKE_BUILD_TYPE=Release -DSM=70,75,80,86 -DBUILD_PYT=ON -DBUILD_MULTI_GPU=ON ..
44+
cmake -DCMAKE_BUILD_TYPE=Release -DSM=70,75,80,86,90 -DBUILD_PYT=ON -DBUILD_MULTI_GPU=ON ..
4545
make -j$(nproc)
4646
cp lib/libth_transformer.so /tmp/binaries/
4747
cd ../../

0 commit comments

Comments
 (0)