sysrap/tests/curand_uniform_test.sh

0001 #!/bin/bash
0002 usage(){ cat << EOU
0003 curand_uniform_test.sh
0004 ========================
0005
0006 ~/o/sysrap/tests/curand_uniform_test.sh
0007
0008 EOU
0009 }
0010
0011 cd $(dirname $(realpath $BASH_SOURCE))
0012 source dbg__.sh
0013
0014 name=curand_uniform_test
0015 src=$name.cu
0016 script=$name.py
0017
0018 tmp=/data/$USER/opticks
0019 export TMP=${TMP:-$tmp}
0020 export FOLD=$TMP/$name
0021 mkdir -p $FOLD
0022 bin=$FOLD/$name
0023
0024 defarg="info_build_run_ana"
0025 arg=${1:-$defarg}
0026
0027 cuda_prefix=/usr/local/cuda
0028 CUDA_PREFIX=${CUDA_PREFIX:-$cuda_prefix}
0029
0030 OPT="-use_fast_math -DRNG_PHILITEOX"
0031
0032 M=1000000
0033 ni=$(( 10*M ))
0034 nj=16
0035 export NI=${NI:-$ni}
0036 export NJ=${NJ:-$nj}
0037
0038 vars="BASH_SOURCE name src script bin FOLD OPT NI NJ"
0039
0040
0041 if [ "${arg/info}" != "$arg" ]; then
0042    for var in $vars ; do printf "%20s : %s \n" "$var" "${!var}" ; done
0043 fi
0044
0045 if [ "${arg/build}" != "$arg" ]; then
0046     nvcc $src -std=c++17 $OPT -lcrypto -lssl -I$HOME/np -I..  -I$CUDA_PREFIX/include -o $bin
0047     [ $? -ne 0 ] && echo compilation error && exit 1
0048 fi
0049
0050 if [ "${arg/run}" != "$arg" ]; then
0051     $bin
0052     [ $? -ne 0 ] && echo run  error && exit 2
0053 fi
0054
0055 if [ "${arg/dbg}" != "$arg" ]; then
0056     dbg__ $bin
0057     [ $? -ne 0 ] && echo dbg  error && exit 2
0058 fi
0059
0060 if [ "${arg/pdb}" != "$arg" ]; then
0061     ${IPYTHON:-ipython} --pdb -i $script
0062     [ $? -ne 0 ] && echo dbg error && exit 3
0063 fi
0064
0065 if [ "${arg/ana}" != "$arg" ]; then
0066     python $script
0067     [ $? -ne 0 ] && echo ana error && exit 4
0068 fi
0069
0070 if [ "${arg/grab}" != "$arg" ]; then
0071     rsync -av P:$FOLD/ $FOLD
0072     [ $? -ne 0 ] && echo grab error && exit 4
0073 fi
0074
0075
0076
0077 exit 0
0078