test_gpu.sh 1.9 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
4
5
#cd /opt/dtk/.hip/
#./replace_origin.sh
#cd -
Your Name's avatar
Your Name committed
6

huchen's avatar
huchen committed
7
cd build;
8
export CPLUS_INCLUDE_PATH=${ROCM_PATH}/include:${ROCM_PATH}:/opencl/include
Your Name's avatar
Your Name committed
9
export LD_LIBRARY_PATH=/public/software/compiler/intel-compiler-2017.5.239/mkl/lib/intel64_lin/:$LD_LIBRARY_PATH
10
11
12
13
#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
14
15

export BUILD_DIR=`pwd`
Your Name's avatar
Your Name committed
16
export LD_LIBRARY_PATH=${BUILD_DIR}/faiss/gpu/test/:${BUILD_DIR}/faiss/:${BUILD_DIR}/lib/:$LD_LIBRARY_PATH
17
#export HIP_KERNEL_PRINTF=1
18
#make -j95 VERBOSE=1
19
export HIP_VISIBLE_DEVICES=3
20

Your Name's avatar
Your Name committed
21
#export MKL_DEBUG_CPU_TYPE=5
Your Name's avatar
Your Name committed
22

23
24
#export HIP_LAUNCH_BLOCKING=1
#export HIP_LOG_LEVEL=7
25

26
27
#export AMD_LOG_LEVEL=4
#export AMD_OCL_WAIT_COMMAND=1
28

29
ctest
30
31

#./faiss/gpu/test/TestGpuIndexFlat > hc_test.log 2>&1 
huchen's avatar
huchen committed
32
#./faiss/gpu/test/TestGpuIndexFlat
33
34
35
36
37
38
39
#ctest -R TestGpuIndexFlat.IP_Float32

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

40
41
42
43
44
45
46
#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
47
48
49
50
51
52
53

##./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
54

55
#ctest -V -R TestGpuSelect.test1Warp