From dd55aa8cf5a0747c7875e7338400ac359f2a6083 Mon Sep 17 00:00:00 2001
From: Jutho <Jutho@users.noreply.github.com>
Date: Wed, 30 Oct 2024 01:18:22 +0100
Subject: [PATCH] fix missing allocator in tensorfree (#197)

---
 .github/workflows/ci.yml            | 2 +-
 Project.toml                        | 2 +-
 src/indexnotation/postprocessors.jl | 5 +++--
 3 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 9bd3049..9fe1c8f 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -66,7 +66,7 @@ jobs:
           - x64
     steps:
       - uses: actions/checkout@v4
-      - uses: julia-actions/setup-julia@v1
+      - uses: julia-actions/setup-julia@v2
         with:
           version: ${{ matrix.version }}
           arch: ${{ matrix.arch }}
diff --git a/Project.toml b/Project.toml
index ed1aa83..1ce2fc4 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "TensorOperations"
 uuid = "6aa20fa7-93e2-5fca-9bc0-fbd0db3c71a2"
 authors = ["Lukas Devos <lukas.devos@ugent.be>", "Maarten Van Damme <maartenvd1994@gmail.com>", "Jutho Haegeman <jutho.haegeman@ugent.be>"]
-version = "5.1.1"
+version = "5.1.2"
 
 [deps]
 CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
diff --git a/src/indexnotation/postprocessors.jl b/src/indexnotation/postprocessors.jl
index cafc7d5..be5097a 100644
--- a/src/indexnotation/postprocessors.jl
+++ b/src/indexnotation/postprocessors.jl
@@ -98,10 +98,11 @@ end
     insertallocator(ex, allocator)
 
 Insert the allocator argument into the tensor operation and allocation methods `tensoradd!`, 
-`tensortrace!`, `tensorcontract!`, `tensoralloc`, `tensoralloc_add`, and `tensoralloc_contract`.
+`tensortrace!`, `tensorcontract!`, `tensoralloc`, `tensoralloc_add`, `tensoralloc_contract`
+and `tensorfree!`.
 """
 function insertallocator(ex, allocator)
     return insertargument(ex, allocator,
                           (:tensoradd!, :tensortrace!, :tensorcontract!, :tensoralloc,
-                           :tensoralloc_add, :tensoralloc_contract))
+                           :tensoralloc_add, :tensoralloc_contract, :tensorfree!))
 end