fix: Link pytorch build files.

Author: mchav

.github/workflows/haskell-ci.yml

@@ -62,11 +62,29 @@ jobs:
       - name: apt-get install
         run: |
           apt-get update
-          apt-get install -y --no-install-recommends gnupg ca-certificates dirmngr curl git software-properties-common libtinfo6 libncurses-dev
+          apt-get install -y --no-install-recommends gnupg ca-certificates dirmngr curl git software-properties-common libtinfo6 libncurses-dev python3 sed
           export LIBTORCH_HOME=$XDG_CACHE_HOME/libtorch
           echo "XDG_CACHE_HOME=$XDG_CACHE_HOME" >> "$GITHUB_ENV"
           echo "LD_LIBRARY_PATH=$LD_LIBRARY_PATH" >> "$GITHUB_ENV"
           echo "LIBTORCH_HOME=$LIBTORCH_HOME" >> "$GITHUB_ENV"
+          export PYTHON=python3
+          git clone https://github.com/pytorch/pytorch.git
+          pushd ./pytorch
+          export VERSION=2.5.0
+          git pull origin v$VERSION
+          git checkout v$VERSION
+          mkdir build
+          $PYTHON -m pip install --user pyyaml
+          $PYTHON -m pip install --user dataclasses
+          $PYTHON -m pip install --user typing_extensions
+          $PYTHON -m torchgen.gen -s aten/src/ATen -d build/aten/src/ATen
+          sed -i -e "s/::std::/std::/g" build/aten/src/ATen/Declarations.yaml
+          sed -i -e "s/ name: n$/ name: 'n'/g" -e "s/ name: N$/ name: 'N'/g" build/aten/src/ATen/Declarations.yaml
+          sed -i -e "s/ name: t$/ name: 't'/g" -e "s/ name: T$/ name: 'T'/g" build/aten/src/ATen/Declarations.yaml
+          sed -i -e "s/ name: y$/ name: 'y'/g" -e "s/ name: Y$/ name: 'Y'/g" build/aten/src/ATen/Declarations.yaml
+          sed -i -e "s/ default: \([^'].*\)$/ default: '\1'/g" build/aten/src/ATen/Declarations.yaml
+          popd
+          ln -fs ./pytorch/build/aten/src/ATen/Declarations.yaml
       - name: Install GHCup
         run: |
           mkdir -p "$HOME/.ghcup/bin"