#!/bin/bash

echo "Copying adaptor files from MindSpeed MM to MindSpeed/Megatron/MindSpeed-RL"
echo "[Warning] Some files will be overwritten in MindSpeed/Megatron/MindSpeed-RL"

while true
do
    read -r -p "Are You Sure? [Y/n] " input

    case $input in
        [yY][eE][sS]|[yY])
            echo "Copying ..."
            break
            ;;

        [nN][oO]|[nN])
            echo "Exit"
            exit 1
            ;;

        *)
            echo "Invalid input..."
            ;;
    esac
done

cp examples/rl/code/build_tokenizer.py mindspeed/features_manager/tokenizer/build_tokenizer.py
cp examples/rl/code/dot_product_attention.py mindspeed/core/context_parallel/dot_product_attention.py
cp examples/rl/code/adaptor.py mindspeed/core/transformer/flash_attention/flash_attention/adaptor.py
cp examples/rl/code/random.py megatron/core/tensor_parallel/random.py
cp examples/rl/code/actor_hybrid_worker.py mindspeed_rl/workers/actor_hybrid_worker.py

echo "All adaptor files copied successfully!"
echo "Total files copied: 5"