compile.sh 1.49 KB
Newer Older
huchen's avatar
huchen committed
1
source /opt/dtk/env.sh
Your Name's avatar
Your Name committed
2

huchen's avatar
huchen committed
3
cd /opt/dtk/.hip/
Your Name's avatar
Your Name committed
4
5
6
7
./remove_origin.sh

cd -

Your Name's avatar
Your Name committed
8
mkdir build -p
9
cd build
huchen's avatar
huchen committed
10
rm hc_* -f
11
12
13
14
15
export CPLUS_INCLUDE_PATH=${ROCM_PATH}/include:${ROCM_PATH}:/opencl/include
export CPLUS_INCLUDE_PATH=${ROCM_PATH}/hiprand/include/:${ROCM_PATH}/rocrand/include:$CPLUS_INCLUDE_PATH
export LD_LIBRARY_PATH=${ROCM_PATH}/hipblas/lib/:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=${ROCM_PATH}/llvm/lib:$LD_LIBRARY_PATH

huchen's avatar
huchen committed
16
#alias python='python3'
huchen's avatar
huchen committed
17
18

export BUILD_DIR=`pwd`
Your Name's avatar
Your Name committed
19
20

#######################
21
#CXX=hipcc CC=hipcc  cmake -DFAISS_ENABLE_GPU=ON -DBUILD_SHARED_LIBS=ON -DBUILD_TESTING=ON -DCMAKE_BUILD_TYPE=DEBUG -DCMAKE_MODULE_PATH="${ROCM_PATH}/hip/cmake/" -DMKL_LIBRARIES=/public/software/compiler/intel-compiler-2017.5.239/mkl/lib/intel64_lin/ ..
huchen's avatar
huchen committed
22
CXX=hipcc CC=hipcc  cmake -DFAISS_ENABLE_GPU=ON -DBUILD_SHARED_LIBS=ON -DFAISS_OPT_LEVEL=avx -DBUILD_TESTING=ON -DCMAKE_BUILD_TYPE=RELEASE -DCMAKE_MODULE_PATH="${ROCM_PATH}/hip/cmake/"  -DCMAKE_INSTALL_PREFIX=${BUILD_DIR}/install  -DMKL_LIBRARIES=/public/software/compiler/intel-compiler-2017.5.239/mkl/lib/intel64_lin/libmkl_rt.so ..
23

24
25
make -j32 faiss VERBOSE=1 > hc_faiss.log 2>&1 
make -j32 swigfaiss VERBOSE=1 > hc_swigfaiss.log 2>&1 
huchen's avatar
huchen committed
26

Your Name's avatar
Your Name committed
27
export LD_LIBRARY_PATH=${BUILD_DIR}/faiss/gpu/test/:${BUILD_DIR}/faiss/:${BUILD_DIR}/lib/:$LD_LIBRARY_PATH
28
29
30
make -j32 install VERBOSE=1 > hc_install.log 2>&1 

#make -j32 test VERBOSE=1 > hc_test.log 2>&1
huchen's avatar
huchen committed
31
cd -
huchen's avatar
huchen committed
32

huchen's avatar
huchen committed
33
#./test_gpu.sh
34

Your Name's avatar
Your Name committed
35
cd build/faiss/python
huchen's avatar
huchen committed
36
37
python3 setup.py -v bdist_wheel
pip3 install dist/* --no-deps --force
Your Name's avatar
Your Name committed
38
39
40
cd -

./test_gpu_python.sh