set_env.sh

rocblas_lib_path=$1

export MIGRAPHX_ENABLE_MIOPEN_GROUPNORM=1
export MIGRAPHX_ENABLE_NHWC=1
export MIGRAPHX_ENABLE_MIOPEN_CONCAT=1
export MIGRAPHX_STABLEDIFFUSION_OPT=1
export MIGRAPHX_ENABLE_MIOPEN_GN_LN=1
export MIGRAPHX_ENABLE_LAYERNORM_FUSION=1
# export PADDING_MALLOC=0  # run on KME

export HIP_VISIBLE_DEVICES=6
export LD_LIBRARY_PATH=${rocblas_lib_path}:$LD_LIBRARY_PATH