install.sh.in

#!/bin/bash

CC=@CLANG_PATH@
CXX=@CLANGXX_PATH@
LINK=@LLVMLINK_PATH@

BAZEL=@BAZEL@

prefix=@prefix@
exec_prefix=@exec_prefix@
INSTALL_LIB_PREFIX=@libdir@
INSTALL_INCLUDE_PREFIX=@includedir@

# Build highway library
$BAZEL build @hwy
HIGHWAY_SHARED_LIB=$(find bazel-bin/ -name libhwy.so)
HIGHWAY_STATIC_LIB=$(find bazel-bin/ -name libhwy.a)

echo "Copying libhwy to $INSTALL_LIB_PREFIX"
chmod u+w $HIGHWAY_SHARED_LIB $HIGHWAY_STATIC_LIB
cp $HIGHWAY_SHARED_LIB $HIGHWAY_STATIC_LIB $INSTALL_LIB_PREFIX

# Build prism library
$BAZEL build src:prism-all

echo "Copying libprism* to $INSTALL_LIB_PREFIX"
chmod u+w bazel-bin/src/libprism*.{a,so}
cp bazel-bin/src/libprism*.{a,so} $INSTALL_LIB_PREFIX

HWY_INCLUDE_PATH=$(${BAZEL} query @hwy --output=location --noshow_loading_progress 2>/dev/null | cut -f1 -d' ' | xargs -n1 dirname)

if [ -z "$HWY_INCLUDE_PATH" ]; then
    echo "Could not find hwy"
    exit 1
fi

# Build LLVM IR prism files
INCLUDE="-I. -I$HWY_INCLUDE_PATH"
CXXFLAGS="-S -emit-llvm -O2 -UPRISM_DEBUG -std=c++17"

# Baseline target
$CXX $CXXFLAGS $INCLUDE -DPRISM_DISPATCH=dynamic_dispatch src/ud_scalar_dynamic.cpp -o ud_scalar_dynamic.ll &
$CXX $CXXFLAGS $INCLUDE -DPRISM_DISPATCH=dynamic_dispatch src/sr_scalar_dynamic.cpp -o sr_scalar_dynamic.ll &

$CXX $CXXFLAGS $INCLUDE -DPRISM_DISPATCH=dynamic_dispatch src/ud_vector_dynamic.cpp -o ud_vector_dynamic.ll &
$CXX $CXXFLAGS $INCLUDE -DPRISM_DISPATCH=dynamic_dispatch src/sr_vector_dynamic.cpp -o sr_vector_dynamic.ll &

# Native target
CXXFLAGS+=" -march=native -DHWY_COMPILE_ONLY_STATIC "

$CXX $CXXFLAGS $INCLUDE -DPRISM_DISPATCH=static_dispatch src/ud_scalar_static.cpp -o ud_scalar_static.ll &
$CXX $CXXFLAGS $INCLUDE -DPRISM_DISPATCH=static_dispatch src/sr_scalar_static.cpp -o sr_scalar_static.ll &

$CXX $CXXFLAGS $INCLUDE -DPRISM_DISPATCH=static_dispatch src/ud_vector_static.cpp -o ud_vector_static.ll &
$CXX $CXXFLAGS $INCLUDE -DPRISM_DISPATCH=static_dispatch src/sr_vector_static.cpp -o sr_vector_static.ll &

wait

$LINK -S {ud,sr}_{scalar,vector}_dynamic.ll -o prism-dynamic.ll
$LINK -S {ud,sr}_{scalar,vector}_static.ll -o prism-static.ll

echo "Copying headers to $INSTALL_INCLUDE_PREFIX"
cp prism-dynamic.ll prism-static.ll $INSTALL_INCLUDE_PREFIX