#!/bin/bash

module purge
module load compiler/devtoolset/7.3.1
module load mpi/hpcx/2.4.1/gcc-7.3.1
module load compiler/rocm/dtk-21.04

#for mlperf_bert
export PYTHONPATH=/path_to_env
export PATH=/path_to_env/bin:$PATH
export LD_LIBRARY_PATH=/path_to_env/lib/python3.6:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/path_to_env/lib:$LD_LIBRARY_PATH
export INCLUDE=/path_to_env/include/python3.6m:$INCLUDE
export ROCM_DIR=/public/software/compiler/rocm/dtk-21.04

export PATH=${ROCM_DIR}/bin:${ROCM_DIR}/hcc/bin:${ROCM_DIR}/hip/bin:${ROCM_DIR}/miopen/bin:${ROCM_DIR}/opencl/bin/x86_64:$PATH
export LD_LIBRARY_PATH=${ROCM_DIR}/lib:${ROCM_DIR}/lib64:${ROCM_DIR}/miopen/lib:${ROCM_DIR}/opencl/lib/x86_64:/public/software/compiler/rocm/dtk-21.04/roctracer/lib:$LD_LIBRARY_PATH
export C_INCLUDE_PATH=${ROCM_DIR}/include:${ROCM_DIR}/opencl/include:${ROCM_DIR}/hiprand/include:${ROCM_DIR}/miopen/include:$C_INCLUDE_PATH
export CPLUS_INCLUDE_PATH=${ROCM_DIR}/include:${ROCM_DIR}/miopen/include:$CPLUS_INCLUDE_PATH
export ROCM_PATH=${ROCM_DIR}
export HIP_PATH=${ROCM_DIR}/hip/


export C_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch/gflags-2.1.2-build/include:/public/software/apps/DeepLearning/PyTorch/glog-build/include:$C_INCLUDE_PATH
export CPLUS_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch/gflags-2.1.2-build/include:/public/software/apps/DeepLearning/PyTorch/glog-build/include:$CPLUS_INCLUDE_PATH
export LD_LIBRARY_PATH=/public/software/apps/DeepLearning/PyTorch/glog-build/lib/:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/public/software/apps/DeepLearning/PyTorch/lmdb-0.9.24-build/lib/:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/public/software/apps/DeepLearning/PyTorch/opencv-2.4.13.6-build/lib/:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/public/software/apps/DeepLearning/PyTorch/openblas-0.3.7-build/lib/:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/public/software/apps/DeepLearning/PyTorch/gflags-2.1.2-build/lib/:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/public/software/apps/DeepLearning/PyTorch/lib/:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/public/software/apps/DeepLearning/PyTorch/openmp-build/lib:$LD_LIBRARY_PATH