test_gpu.sh 1.83 KB
Newer Older
1
source /opt/dtk-22.04.2/env.sh
huchen's avatar
huchen committed
2
cd build;
3
4
5
6
7
8
export CPLUS_INCLUDE_PATH=${ROCM_PATH}/include:${ROCM_PATH}:/opencl/include
export LD_LIBRARY_PATH=/opt/intel/compilers_and_libraries_2018.1.163/linux/mkl/lib/intel64_lin/:$LD_LIBRARY_PATH
#export PATH=/home/huchen/FAISS/swig-4.0.2-build/bin/:$PATH
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
Your Name's avatar
Your Name committed
9
10
11

export BUILD_DIR=`pwd`
export LD_LIBRARY_PATH=${BUILD_DIR}/faiss/gpu/test/:${BUILD_DIR}faiss/:${BUILD_DIR}/lib/:$LD_LIBRARY_PATH
12
#export HIP_KERNEL_PRINTF=1
13
#make -j95 VERBOSE=1
14
export HIP_VISIBLE_DEVICES=3
15

16
17
#export HIP_LAUNCH_BLOCKING=1
#export HIP_LOG_LEVEL=7
18

19
20
#export AMD_LOG_LEVEL=4
#export AMD_OCL_WAIT_COMMAND=1
21

22
ctest
23
24

#./faiss/gpu/test/TestGpuIndexFlat > hc_test.log 2>&1 
huchen's avatar
huchen committed
25
#./faiss/gpu/test/TestGpuIndexFlat
26
27
28
29
30
31
32
#ctest -R TestGpuIndexFlat.IP_Float32

#ctest -R TestGpuIndexIVFFlat.Float32_32_Add_L2
#./faiss/gpu/test/TestGpuIndexIVFFlat
#./faiss/gpu/test/TestGpuIndexIVFPQ
#./faiss/gpu/test/TestGpuIndexIVFScalarQuantizer

33
34
35
36
37
38
39
#ctest -R TestGpuIndexIVFPQ
#ctest -R TestGpuIndexFlat

#gdb --args ./faiss/gpu/test/TestGpuIndexIVFPQ --gtest_filter=TestGpuIndexIVFPQ.Query_L2_MMCodeDistance
#./faiss/gpu/test/TestGpuIndexIVFPQ --gtest_filter=TestGpuIndexIVFPQ.Query_L2_MMCodeDistance
#./faiss/gpu/test/TestGpuIndexIVFPQ --gtest_filter=TestGpuIndexIVFPQ.Query_IP_MMCodeDistance
#./faiss/gpu/test/TestGpuIndexIVFPQ --gtest_filter=TestGpuIndexIVFPQ.Query_IP_MMCodeDistance > hc_test.log 2>&1
40
41
42
43
44
45
46

##./faiss/gpu/test/TestGpuIndexFlat
##./faiss/gpu/test/TestCodePacking
##./faiss/gpu/test/TestGpuIndexBinaryFlat
##./faiss/gpu/test/TestGpuMemoryException
##./faiss/gpu/test/TestGpuSelect
##./faiss/gpu/test/TestGpuDistance
huchen's avatar
huchen committed
47

48
#ctest -V -R TestGpuSelect.test1Warp