build.sh

current_path=$(pwd)
conda_path=$HOME/miniconda3/envs

for dir in pytorch_cluster pytorch_scatter pytorch_sparse pytorch_spline_conv
do
  (
    cd "${current_path}/$dir"
    ${conda_path}/py39/bin/python setup.py bdist_wheel &
    ${conda_path}/py310/bin/python setup.py bdist_wheel
  ) &
done

wait

for dir in pytorch_cluster pytorch_scatter pytorch_sparse pytorch_spline_conv
do
  (
    cd "${current_path}/$dir"
    ${conda_path}/py311/bin/python setup.py bdist_wheel &
    ${conda_path}/py312/bin/python setup.py bdist_wheel
  ) &
done

wait

for dir in pytorch_cluster pytorch_scatter pytorch_sparse pytorch_spline_conv
do
  (
    cd "${current_path}/$dir"
    ${conda_path}/py38/bin/python setup.py bdist_wheel
  ) &
done

wait

if [ -d ${current_path}/dist ]; then
  rm -rf ${current_path}/dist
fi
mkdir ${current_path}/dist

package_name="torch-2.4-rocm-6.1"

versions=("py38" "py39" "py310" "py311" "py312")

for version in "${versions[@]}"; do
    mkdir -p ${current_path}/dist/${package_name}-${version}-linux_x86_64
    
    for module in "pytorch_cluster" "pytorch_scatter" "pytorch_sparse" "pytorch_spline_conv"; do
        mv ${current_path}/${module}/dist/*-cp${version: -2}-cp${version: -2}-linux_x86_64.whl ${current_path}/dist/${package_name}-${version}-linux_x86_64/
    done

    zip ${current_path}/dist/${package_name}-${version}-linux_x86_64.zip ${current_path}/dist/${package_name}-${version}-linux_x86_64/*.whl
done

for module in "pytorch_cluster" "pytorch_scatter" "pytorch_sparse" "pytorch_spline_conv"; do
    rm -rf ${current_path}/${module}/dist
    rm -rf ${current_path}/${module}/build
done


wget "https://files.pythonhosted.org/packages/03/9f/157e913626c1acfb3b19ce000b1a6e4e4fb177c0bc0ea0c67ca5bd714b5a/torch_geometric-2.6.1-py3-none-any.whl" -O "${current_path}/dist/torch_geometric-2.6.1-py3-none-any.whl"

for version in "${versions[@]}"; do
    cp ${current_path}/dist/torch_geometric-2.6.1-py3-none-any.whl ${current_path}/dist/${package_name}-${version}-linux_x86_64/
    zip ${current_path}/dist/${package_name}-${version}-linux_x86_64.zip ${current_path}/dist/${package_name}-${version}-linux_x86_64/*.whl
done

rm ${current_path}/dist/torch_geometric-2.6.1-py3-none-any.whl