Skip to content

Commit

Permalink
fix missing allocator in tensorfree
Browse files Browse the repository at this point in the history
  • Loading branch information
Jutho committed Oct 29, 2024
1 parent 0135c8f commit d33b572
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 4 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/ci.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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 }}
Expand Down
2 changes: 1 addition & 1 deletion Project.toml
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
name = "TensorOperations"
uuid = "6aa20fa7-93e2-5fca-9bc0-fbd0db3c71a2"
authors = ["Lukas Devos <[email protected]>", "Maarten Van Damme <[email protected]>", "Jutho Haegeman <[email protected]>"]
version = "5.1.1"
version = "5.1.2"

[deps]
CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
Expand Down
5 changes: 3 additions & 2 deletions src/indexnotation/postprocessors.jl
Original file line number Diff line number Diff line change
Expand Up @@ -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

0 comments on commit d33b572

Please sign in to comment.