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=/public/home/zhuww/wangwf/pkgs/rocblas-install-0626/lib:$LD_LIBRARY_PATH
# export LD_LIBRARY_PATH=${rocblas_lib_path}:$LD_LIBRARY_PATH

MIGRAPHX_TRANSPOSE_CONTIGUOUS_GEMM=1