OpenBLAS/c_check

#!/bin/sh

# Checking cross compile
hostos="$(uname -s | sed -e 's/\-.*//')"
hostarch="$(uname -m | sed -e 's/i.86/x86/')"
if [ "$hostos" = "AIX" ] || [ "$hostos" = "SunOS" ]; then
    hostarch="$(uname -p)"
fi
case "$hostarch" in
    amd64) hostarch=x86_64 ;;
    arm*) [ "$hostarch" = "arm64" ] || hostarch='arm' ;;
    aarch64) hostarch=arm64 ;;
    powerpc*|ppc*) hostarch=power ;;
    s390x) hostarch=zarch ;;
esac

makefile="$1"
config="$2"

compiler_name="$3"
shift 3
flags="$*"

# First, we need to know the target OS and compiler name
{
    data=$($compiler_name $flags -E ctest.c)
} || {
    printf '%s\n' "C Compiler ($compiler_name) is something wrong." >&2
    exit 1
}

cross_suffix=""

if [ "$(dirname $compiler_name)" != '.' ]; then
    cross_suffix="$cross_suffix$(dirname $compiler_name)/"
fi

bn="$(basename $compiler_name)"
case "$bn" in
    *-*) cross_suffix="$cross_suffix${bn%-*}-"
esac

compiler=""
case "$data" in
    *COMPILER_LSB*) compiler=LSB ;;
    *COMPILER_CLANG*) compiler=CLANG ;;
    *COMPILER_PGI*) compiler=PGI ;;
    *COMPILER_PATHSCALE*) compiler=PATHSCALE ;;
    *COMPILER_INTEL*) compiler=INTEL ;;
    *COMPILER_OPEN64*) compiler=OPEN64 ;;
    *COMPILER_SUN*) compiler=SUN ;;
    *COMPILER_IBM*) compiler=IBM ;;
    *COMPILER_DEC*) compiler=DEC ;;
esac
if [ -z "$compiler" ]; then
    compiler=GCC
fi

case "$data" in *OS_LINUX*) os=Linux ;; esac
case "$data" in *OS_FREEBSD*) os=FreeBSD ;; esac
case "$data" in *OS_NETBSD*) os=NetBSD ;; esac
case "$data" in *OS_OPENBSD*) os=OpenBSD ;; esac
case "$data" in *OS_DRAGONFLY*) os=DragonFly ;; esac
case "$data" in *OS_DARWIN*) os=Darwin ;; esac
case "$data" in *OS_SUNOS*) os=SunOS ;; esac
case "$data" in *OS_AIX*) os=AIX ;; esac
case "$data" in *OS_OSF*) os=osf ;; esac
case "$data" in *OS_WINNT*) os=WINNT ;; esac
case "$data" in *OS_CYGWIN_NT*) os=CYGWIN_NT ;; esac
case "$data" in *OS_INTERIX*) os=Interix ;; esac
case "$data" in *OS_ANDROID*) os=Android ;; esac
case "$data" in *OS_HAIKU*) os=Haiku ;; esac

case "$data" in
    *ARCH_X86_64*) architecture=x86_64 ;;
    *ARCH_X86*) architecture=x86 ;;
    *ARCH_E2K*) architecture=e2k ;;
    *ARCH_POWER*) architecture=power ;;
    *ARCH_MIPS64*) architecture=mips64 ;;
    *ARCH_MIPS*) architecture=mips ;;
    *ARCH_ALPHA*) architecture=alpha ;;
    *ARCH_SPARC*) architecture=sparc ;;
    *ARCH_IA64*) architecture=ia64 ;;
    *ARCH_ARM64*) architecture=arm64 ;;
    *ARCH_ARM*) architecture=arm ;;
    *ARCH_ZARCH*) architecture=zarch ;;
    *ARCH_RISCV64*) architecture=riscv64 ;;
    *ARCH_LOONGARCH64*) architecture=loongarch64 ;;
esac

defined=0

if [ "$os" = "AIX" ]; then
    case "$BINARY" in
        32) compiler_name="$compiler_name -maix32" ;;
        64) compiler_name="$compiler_name -maix64" ;;
    esac
    defined=1
fi

case "$architecture" in
    mips)
    	compiler_name="$compiler_name -mabi=32"
    	defined=1
    	;;
    mips64)
    	case "$BINARY" in
    	    32) compiler_name="$compiler_name -mabi=n32" ;;
    	    64) compiler_name="$compiler_name -mabi=64" ;;
    	esac
    	defined=1
    	;;
    arm|arm64) defined=1 ;;
    zarch|e2k|alpha|ia64|riscv64|loonarch64)
    	defined=1
    	BINARY=64
    	;;
    x86)
    	[ "$os" != "Darwin" ] && [ "$os" != "SunOS" ] && {
    	    defined=1
    	    BINARY=32
    	}
    	;;
esac

case "$compiler" in
    PGI)
    	case "$BINARY" in
    	    32) compiler_name="$compiler_name -tp p7" ;;
    	    64) compiler_name="$compiler_name -tp p7-64" ;;
    	esac
    	openmp='-mp'
    	defined=1
    	;;
    IBM)
    	case "$BINARY" in
    	    32) compiler_name="$compiler_name -q32" ;;
    	    64) compiler_name="$compiler_name -q64" ;;
    	esac
    	openmp='-qsmp=omp'
    	defined=1
    	;;
    INTEL) openmp='-openmp' ;;
    PATHSCALE|OPEN64) openmp='-mp' ;;
    CLANG|GCC|LSB) openmp='-fopenmp' ;;
esac

if [ "$defined" -eq 0 ]; then
    case "$BINARY" in
    	32) compiler_name="$compiler_name -m32" ;;
    	64) compiler_name="$compiler_name -m64" ;;
    esac
fi

# Do again
{
    data=$($compiler_name $flags -E ctest.c)
} || {
    printf '%s\n' "C Compiler ($compiler_name) is something wrong." >&2
    exit 1
}

have_msa=0
if [ "$architecture" = "mips" ] || [ "$architecture" = "mips64" ]; then
    tmpd=$(mktemp -d)
    tmpf="$tmpd/a.c"
    code='"addvi.b $w0, $w1, 1"'
    msa_flags='-mmsa -mfp64 -mload-store-pairs'
    printf "#include <msa.h>\n\n" >> "$tmpf"
    printf "void main(void){ __asm__ volatile(%s); }\n" "$code" >> "$tmpf"

    args="$msa_flags -o $tmpf.o $tmpf"
    have_msa=1
    {
    	$compiler_name $flags $args >/dev/null 2>&1
    } || {
    	have_msa=0
    }

    rm -rf "$tmpd"
fi

case "$data" in
    *ARCH_X86_64*) architecture=x86_64 ;;
    *ARCH_X86*) architecture=x86 ;;
    *ARCH_E2K*) architecture=e2k ;;
    *ARCH_POWER*) architecture=power ;;
    *ARCH_MIPS64*) architecture=mips64 ;;
    *ARCH_MIPS*) architecture=mips ;;
    *ARCH_ALPHA*) architecture=alpha ;;
    *ARCH_SPARC*) architecture=sparc ;;
    *ARCH_IA64*) architecture=ia64 ;;
    *ARCH_ARM64*) architecture=arm64 ;;
    *ARCH_ARM*) architecture=arm ;;
    *ARCH_ZARCH*) architecture=zarch ;;
    *ARCH_LOONGARCH64*) architecture=loongarch64 ;;
esac

binformat='bin32'
case "$data" in
    *BINARY_64*) binformat='bin64' ;;
esac

no_avx512=0
if [ "$architecture" = "x86" ] || [ "$architecture" = "x86_64" ]; then
    tmpd=$(mktemp -d)
    tmpf="$tmpd/a.c"
    code='"vbroadcastss -4 * 4(%rsi), %zmm2"'
    printf "#include <immintrin.h>\n\nint main(void){ __asm__ volatile(%s); }\n" "$code" >> "$tmpf"
    if [ "$compiler" = "PGI" ]; then
        args=" -tp skylake -c -o $tmpf.o $tmpf"
    else
        args=" -march=skylake-avx512 -c -o $tmpf.o $tmpf"
    fi
    no_avx512=0
    {
    	$compiler_name $flags $args >/dev/null 2>&1
    } || {
    	no_avx512=1
    }

    rm -rf "$tmpd"
fi

c11_atomics=0
case "$data" in
    *HAVE_C11*)
        tmpd=$(mktemp -d)
        tmpf="$tmpd/a.c"
        printf "#include <stdatomic.h>\nint main(void){}\n" >> "$tmpf"
        args=' -c -o $tmpf.o $tmpf'
        c11_atomics=1
        {
            $compiler_name $flags $args >/dev/null 2>&1
        } || {
            c11_atomics=0
        }

        rm -rf "$tmpd"
        ;;
esac

if [ "$compiler" = "GCC" ]; then
    case "$architecture" in x86|x86_64)
        no_avx2=0
        oldgcc=0
        data=$($compiler_name -dumpversion)
        case "$data" in *.*.*)
            data="${data%.*}"
        esac
        if awk -v n1=$data -v n2=4.6 'BEGIN { exit !(n1 <= n2) }'; then
            no_avx2=1
            oldgcc=1
        fi
    esac
fi

data="$($compiler_name $flags -S ctest1.c && grep globl ctest1.s | head -n 1 && rm -f ctest1.s)"

need_fu=''
if echo "$data" | grep -q 'globl[[:space:]][_\.]'; then
    need_fu="${data##*globl[[:space:]]}"
    need_fu="${need_fu%%[!_\.]*}"
fi

cross=0

if [ "$architecture" != "$hostarch" ]; then
    cross=1
    [ "$hostarch" = "x86_64" ] && [ "$architecture" = "x86" ] && cross=0
    [ "$hostarch" = "mips64" ] && [ "$architecture" = "mips" ] && cross=0
fi

[ "$os" != "$hostos" ] && cross=1
[ "$os" = "Android" ] && [ "$hostos" = "Linux" ] && [ -n "$TERMUX_APP_PID" ] \
    && cross=0

[ "$USE_OPENMP" != 1 ] && openmp=''

linker_L=""
linker_l=""
linker_a=""

link="$($compiler_name $flags -c ctest2.c -o ctest2.o 2>&1 && $compiler_name $flags $openmp -v ctest2.o -o ctest2 2>&1 && rm -f ctest2.o ctest2 ctest2.exe)"

link="$(echo "$link" | sed 's/\-Y[[:space:]]P\,/\-Y/g')"

flags="$(echo "$link" | tr '[:space:],\n' ' ')"

# Strip trailing quotes
old_flags="$flags"
flags=''
for flag in $old_flags; do
    f="${flag#\"}"
    f="${flag%\"}"
    flags="$flags $f"
done

for flag in $flags; do
    case "$flag" in -L*)
        case "$flag" in
            -LIST:*|-LANG:*) ;;
            *) linker_L="$linker_L $flag" ;;
        esac
    esac

    case "$flag" in -Y*)
        linker_L="$linker_L -Wl,$flag" ;;
    esac

    case "$flag" in --exclude-libs*)
        linker_L="$linker_L -Wl,$flag"
        flags=""
        ;;
    esac

    case "$flag" in -l*)
        case "$flag" in
            *gfortranbegin*|*frtbegin*|*pathfstart*|*numa*|*crt[0-9]*|\
                *gcc*|*user32*|*kernel32*|*advapi32*|*shell32*|*omp*|\
                *[0-9]*) ;;
            *) linker_l="$linker_l $flag" ;;
        esac
    esac

    case "$flag" in *.a) linker_a="$linker_a $flag" ;; esac
done

:> "$makefile" || exit 1
:> "$config" || exit 1

# print $data, "\n";

printf "OSNAME=%s\n" "$os" >> "$makefile"
printf "ARCH=%s\n" "$architecture" >> "$makefile"
printf "C_COMPILER=%s\n" "$compiler" >> "$makefile"
[ "$binformat" = "bin32" ] && printf "BINARY32=\n"            >> "$makefile"
[ "$binformat" = "bin64" ] && printf "BINARY64=\n"            >> "$makefile"
[ "$binformat" = "bin32" ] && printf "BINARY32=1\n"           >> "$makefile"
[ "$binformat" = "bin64" ] && printf "BINARY64=1\n"           >> "$makefile"
[ -n "$need_fu" ] && printf 'FU=%s\n' "$need_fu"                 >> "$makefile"
[ "$cross" -ne 0 ] && [ -n "$cross_suffix" ] && \
    printf "CROSS_SUFFIX=%s\n" "$cross_suffix"                   >> "$makefile"
[ "$cross" -ne 0 ] && printf "CROSS=1\n"                         >> "$makefile"
printf "CEXTRALIB=%s %s %s\n" "$linker_L" "$linker_l" "$linker_a" >> "$makefile"
[ "$have_msa" -eq 1 ] && {
    printf "HAVE_MSA=1\n"                                        >> "$makefile"
    printf "MSA_FLAGS=%s\n" "$msa_flags"                         >> "$makefile"
}
! [ "$no_avx512" -eq 1 ] || printf "NO_AVX512=1\n"               >> "$makefile"
! [ "$no_avx2" -eq 1 ] || printf "NO_AVX2=1\n"                   >> "$makefile"
! [ "$oldgcc" -eq 1 ] || printf "OLDGCC=1\n"                     >> "$makefile"

os="$(echo "$os" | tr '[:lower:]' '[:upper:]'/ )"
architecture="$(echo "$architecture" | tr '[:lower:]' '[:upper:]' )"
compiler="$(echo "$compiler" | tr '[:lower:]' '[:upper:]' )"

{
    printf "#define OS_%s\t1\n" "$os"
    printf "#define ARCH_%s\t1\n" "$architecture"
    printf "#define C_%s\t1\n" "$compiler"
    [ "$binformat" = "bin32" ] && printf "#define __32BIT__\t1\n"
    [ "$binformat" = "bin64" ] && printf "#define __64BIT__\t1\n"
    [ -n "$need_fu" ] && printf "#define FUNDERSCORE\t%s\n" "$need_fu"
    ! [ "$have_msa" -eq 1 ] || printf "#define HAVE_MSA\t1\n"
    ! [ "$c11_atomics" -eq 1 ] || printf "#define HAVE_C11\t1\n"
} >> "$config"


if [ "$os" = "LINUX" ]; then

#    @pthread = split(/\s+/, `nm /lib/libpthread.so* | grep _pthread_create`);

#    if ($pthread[2] ne "") {
#	print CONFFILE "#define PTHREAD_CREATE_FUNC	$pthread[2]\n";
#    } else {
	printf "#define PTHREAD_CREATE_FUNC	pthread_create\n" >> "$config"
#    }
else
    printf "#define PTHREAD_CREATE_FUNC	pthread_create\n" >> "$config"
fi