-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathset_env.sh
executable file
·38 lines (29 loc) · 971 Bytes
/
set_env.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
#!/bin/bash
export GSKNN_DIR=$PWD
echo "GSKNN_DIR = $GSKNN_DIR"
## Manually set the target architecture.
export GSKNN_ARCH_MAJOR=x86_64
export GSKNN_ARCH_MINOR=sandybridge
#export GSKNN_ARCH_MAJOR=x86_64
#export GSKNN_ARCH_MINOR=haswell
#export GSKNN_ARCH_MAJOR=arm
#export GSKNN_ARCH_MINOR=neon
export GSKNN_ARCH=$GSKNN_ARCH_MAJOR/$GSKNN_ARCH_MINOR
echo "GSKNN_ARCH = $GSKNN_ARCH"
## For Macbook Pro
export DYLD_LIBRARY_PATH=$DYLD_LIBRARY_PATH:/opt/intel/lib:/opt/intel/mkl/lib
echo "DYLD_LIBRARY_PATH = $DYLD_LIBRARY_PATH"
## Compiler options (if false, then use GNU compilers)
#export GSKNN_USE_INTEL=true
#echo "GSKNN_USE_INTEL = $GSKNN_USE_INTEL"
#export GSKNN_USE_GNU=false
#echo "GSKNN_USE_GNU = $GSKNN_USE_GNU"
## Whether use BLAS or not?
export GSKNN_USE_BLAS="True"
echo "GSKNN_USE_BLAS = $GSKNN_USE_BLAS"
## Manually set the mkl path/hint
export MKLROOT=""
## Parallel options
export KMP_AFFINITY=compact
export OMP_NUM_THREADS=2
export GSKNN_IC_NT=2