#!/bin/bash

BUILD_DIR=$1
if [ "" = "$BUILD_DIR" ]; then
    BUILD_DIR='./build'
fi

mkdir -p $BUILD_DIR

python3 -c 'import tensorflow as tf; print(" ".join(tf.sysconfig.get_compile_flags()))' > $BUILD_DIR/TF_CFLAGS
python3 -c 'import tensorflow as tf; print(" ".join(tf.sysconfig.get_link_flags()))' > $BUILD_DIR/TF_LFLAGS