ROCm support #3462
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,212 @@ | ||
| #!/bin/bash | ||
|
|
||
| # go one level up from faiss/gpu | ||
| top=$(dirname "${BASH_SOURCE[0]}")/.. | ||
| echo "top=$top" | ||
| cd $top | ||
| echo "pwd=`pwd`" | ||
|
|
||
| # create all destination directories for hipified files into sibling 'gpu-rocm' directory | ||
| for src in $(find ./gpu -type d) | ||
| do | ||
| dst=$(echo $src | sed 's/gpu/gpu-rocm/') | ||
| echo "Creating $dst" | ||
| mkdir -p $dst | ||
| done | ||
|
|
||
| # run hipify-perl against all *.cu *.cuh *.h *.cpp files, no renaming | ||
| # run all files in parallel to speed up | ||
| for ext in cu cuh h cpp | ||
| do | ||
| for src in $(find ./gpu -name "*.$ext") | ||
| do | ||
| dst=$(echo $src | sed 's@./gpu@./gpu-rocm@') | ||
| hipify-perl -o=$dst.tmp $src & | ||
| done | ||
| done | ||
| wait | ||
|
|
||
| # rename all hipified *.cu files to *.hip | ||
| for src in $(find ./gpu-rocm -name "*.cu.tmp") | ||
| do | ||
| dst=${src%.cu.tmp}.hip.tmp | ||
| mv $src $dst | ||
| done | ||
|
|
||
| # replace header include statements "<faiss/gpu/" with "<faiss/gpu-rocm" | ||
| # replace thrust::cuda::par with thrust::hip::par | ||
| # adjust header path location for hipblas.h to avoid unnecessary deprecation warnings | ||
| # adjust header path location for hiprand_kernel.h to avoid unnecessary deprecation warnings | ||
| for ext in hip cuh h cpp | ||
| do | ||
| for src in $(find ./gpu-rocm -name "*.$ext.tmp") | ||
| do | ||
| sed -i 's@#include <faiss/gpu/@#include <faiss/gpu-rocm/@' $src | ||
| sed -i 's@thrust::cuda::par@thrust::hip::par@' $src | ||
| sed -i 's@#include <hipblas.h>@#include <hipblas/hipblas.h>@' $src | ||
| sed -i 's@#include <hiprand_kernel.h>@#include <hiprand/hiprand_kernel.h>@' $src | ||
| done | ||
| done | ||
|
|
||
| # hipify was run in parallel above | ||
| # don't copy the tmp file if it is unchanged | ||
| for ext in hip cuh h cpp | ||
| do | ||
| for src in $(find ./gpu-rocm -name "*.$ext.tmp") | ||
| do | ||
| dst=${src%.tmp} | ||
| if test -f $dst | ||
| then | ||
| if diff -q $src $dst >& /dev/null | ||
| then | ||
| echo "$dst [unchanged]" | ||
| rm $src | ||
| else | ||
| echo "$dst" | ||
| mv $src $dst | ||
| fi | ||
| else | ||
| echo "$dst" | ||
| mv $src $dst | ||
| fi | ||
| done | ||
| done | ||
|
|
||
| # copy over CMakeLists.txt | ||
| for src in $(find ./gpu -name "CMakeLists.txt") | ||
| do | ||
| dst=$(echo $src | sed 's@./gpu@./gpu-rocm@') | ||
| if test -f $dst | ||
| then | ||
| if diff -q $src $dst >& /dev/null | ||
| then | ||
| echo "$dst [unchanged]" | ||
| else | ||
| echo "$dst" | ||
| cp $src $dst | ||
| fi | ||
| else | ||
| echo "$dst" | ||
| cp $src $dst | ||
| fi | ||
| done | ||
|
|
||
| # Copy over other files | ||
| for ext in py | ||
| do | ||
| for src in $(find ./gpu -name "*.$ext") | ||
| do | ||
| dst=$(echo $src | sed 's@./gpu@./gpu-rocm@') | ||
| if test -f $dst | ||
| then | ||
| if diff -q $src $dst >& /dev/null | ||
| then | ||
| echo "$dst [unchanged]" | ||
| else | ||
| echo "$dst" | ||
| cp $src $dst | ||
| fi | ||
| else | ||
| echo "$dst" | ||
| cp $src $dst | ||
| fi | ||
| done | ||
| done | ||
|
|
||
|
|
||
| ################################################################################### | ||
| # C_API Support | ||
| ################################################################################### | ||
|
|
||
| # Now get the c_api dir | ||
| # This points to the faiss/c_api dir | ||
| top_c_api=$(dirname "${BASH_SOURCE[0]}")/../../c_api | ||
| echo "top=$top_c_api" | ||
| cd ../$top_c_api | ||
| echo "pwd=`pwd`" | ||
|
|
||
|
|
||
| # create all destination directories for hipified files into sibling 'gpu-rocm' directory | ||
| for src in $(find ./gpu -type d) | ||
| do | ||
| dst=$(echo $src | sed 's/gpu/gpu-rocm/') | ||
| echo "Creating $dst" | ||
| mkdir -p $dst | ||
| done | ||
|
|
||
| # run hipify-perl against all *.cu *.cuh *.h *.cpp files, no renaming | ||
| # run all files in parallel to speed up | ||
| for ext in cu cuh h cpp c | ||
| do | ||
| for src in $(find ./gpu -name "*.$ext") | ||
| do | ||
| dst=$(echo $src | sed 's@./gpu@./gpu-rocm@') | ||
| hipify-perl -o=$dst.tmp $src & | ||
| done | ||
| done | ||
| wait | ||
|
|
||
| # rename all hipified *.cu files to *.hip | ||
| for src in $(find ./gpu-rocm -name "*.cu.tmp") | ||
| do | ||
| dst=${src%.cu.tmp}.hip.tmp | ||
| mv $src $dst | ||
| done | ||
|
|
||
| # replace header include statements "<faiss/gpu/" with "<faiss/gpu-rocm" | ||
| # replace thrust::cuda::par with thrust::hip::par | ||
| # adjust header path location for hipblas.h to avoid unnecessary deprecation warnings | ||
| # adjust header path location for hiprand_kernel.h to avoid unnecessary deprecation warnings | ||
| for ext in hip cuh h cpp c | ||
| do | ||
| for src in $(find ./gpu-rocm -name "*.$ext.tmp") | ||
| do | ||
| sed -i 's@#include <faiss/gpu/@#include <faiss/gpu-rocm/@' $src | ||
| sed -i 's@thrust::cuda::par@thrust::hip::par@' $src | ||
| sed -i 's@#include <hipblas.h>@#include <hipblas/hipblas.h>@' $src | ||
| sed -i 's@#include <hiprand_kernel.h>@#include <hiprand/hiprand_kernel.h>@' $src | ||
| done | ||
| done | ||
|
|
||
| # hipify was run in parallel above | ||
| # don't copy the tmp file if it is unchanged | ||
| for ext in hip cuh h cpp c | ||
| do | ||
| for src in $(find ./gpu-rocm -name "*.$ext.tmp") | ||
| do | ||
| dst=${src%.tmp} | ||
| if test -f $dst | ||
| then | ||
| if diff -q $src $dst >& /dev/null | ||
| then | ||
| echo "$dst [unchanged]" | ||
| rm $src | ||
| else | ||
| echo "$dst" | ||
| mv $src $dst | ||
| fi | ||
| else | ||
| echo "$dst" | ||
| mv $src $dst | ||
| fi | ||
| done | ||
| done | ||
|
|
||
| # copy over CMakeLists.txt | ||
| for src in $(find ./gpu -name "CMakeLists.txt") | ||
| do | ||
| dst=$(echo $src | sed 's@./gpu@./gpu-rocm@') | ||
| if test -f $dst | ||
| then | ||
| if diff -q $src $dst >& /dev/null | ||
| then | ||
| echo "$dst [unchanged]" | ||
| else | ||
| echo "$dst" | ||
| cp $src $dst | ||
| fi | ||
| else | ||
| echo "$dst" | ||
| cp $src $dst | ||
| fi | ||
| done |
Oops, something went wrong.
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Could you name it something like
GPU_EXT_PREFIX?