test_gpu.sh 1.86 KB
Newer Older
1
source /opt/dtk-22.04.2/env.sh
huchen's avatar
huchen committed
2
cd build;
3
export CPLUS_INCLUDE_PATH=${ROCM_PATH}/include:${ROCM_PATH}:/opencl/include
Your Name's avatar
Your Name committed
4
export LD_LIBRARY_PATH=/public/software/compiler/intel-compiler-2017.5.239/mkl/lib/intel64_lin/:$LD_LIBRARY_PATH
5
6
7
8
#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

Your Name's avatar
Your Name committed
16
17
export MKL_DEBUG_CPU_TYPE=5

18
19
#export HIP_LAUNCH_BLOCKING=1
#export HIP_LOG_LEVEL=7
20

21
22
#export AMD_LOG_LEVEL=4
#export AMD_OCL_WAIT_COMMAND=1
23

24
ctest
25
26

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

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

35
36
37
38
39
40
41
#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
42
43
44
45
46
47
48

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

50
#ctest -V -R TestGpuSelect.test1Warp