#!/bin/bash
#
# Copyright 2023 Intel Corporation.
# 
# This software and the related documents are Intel copyrighted materials, and
# your use of them is governed by the express license under which they were
# provided to you (License). Unless the License provides otherwise, you may
# not use, modify, copy, publish, distribute, disclose or transmit this
# software or the related documents without Intel's prior written permission.
# 
# This software and the related documents are provided as is, with no express
# or implied warranties, other than those that are expressly stated in the
# License.
#

ln -sf $CONDA_PREFIX/share/intel_optimization_for_horovod/lib/libccl.so.1.0 $CONDA_PREFIX/share/intel_optimization_for_horovod/lib/libccl.so.1
ln -sf $CONDA_PREFIX/share/intel_optimization_for_horovod/lib/libccl.so.1.0 $CONDA_PREFIX/share/intel_optimization_for_horovod/lib/libccl.so

ln -sf $CONDA_PREFIX/share/intel_optimization_for_horovod/lib/libmpi.so.12.0.0 $CONDA_PREFIX/share/intel_optimization_for_horovod/lib/libmpi.so.12.0
ln -sf $CONDA_PREFIX/share/intel_optimization_for_horovod/lib/libmpi.so.12.0.0 $CONDA_PREFIX/share/intel_optimization_for_horovod/lib/libmpi.so.12
ln -sf $CONDA_PREFIX/share/intel_optimization_for_horovod/lib/libmpi.so.12.0.0 $CONDA_PREFIX/share/intel_optimization_for_horovod/lib/libmpi.so

ln -sf $CONDA_PREFIX/share/intel_optimization_for_horovod/lib/libmpicxx.so.12.0.0 $CONDA_PREFIX/share/intel_optimization_for_horovod/lib/libmpicxx.so.12.0
ln -sf $CONDA_PREFIX/share/intel_optimization_for_horovod/lib/libmpicxx.so.12.0.0 $CONDA_PREFIX/share/intel_optimization_for_horovod/lib/libmpicxx.so.12
ln -sf $CONDA_PREFIX/share/intel_optimization_for_horovod/lib/libmpicxx.so.12.0.0 $CONDA_PREFIX/share/intel_optimization_for_horovod/lib/libmpicxx.so

ln -sf $CONDA_PREFIX/share/intel_optimization_for_horovod/lib/libmpifort.so.12.0.0 $CONDA_PREFIX/share/intel_optimization_for_horovod/lib/libmpifort.so.12.0
ln -sf $CONDA_PREFIX/share/intel_optimization_for_horovod/lib/libmpifort.so.12.0.0 $CONDA_PREFIX/share/intel_optimization_for_horovod/lib/libmpifort.so.12
ln -sf $CONDA_PREFIX/share/intel_optimization_for_horovod/lib/libmpifort.so.12.0.0 $CONDA_PREFIX/share/intel_optimization_for_horovod/lib/libmpifort.so

source $CONDA_PREFIX/share/intel_optimization_for_horovod/env/setvars.sh

export CCL_ZE_IPC_EXCHANGE=sockets
