Skip to content

Commit

Permalink
deps, CI and @tensor call updates for 1.11
Browse files Browse the repository at this point in the history
  • Loading branch information
lpawela committed Oct 21, 2024
1 parent 970f4f3 commit ff99e08
Show file tree
Hide file tree
Showing 5 changed files with 8 additions and 12 deletions.
1 change: 1 addition & 0 deletions .github/workflows/CI.yml
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@ jobs:
version:
- '1.9'
- '1.10'
- '1.11'
steps:
- uses: actions/checkout@v4
- uses: julia-actions/setup-julia@v2
Expand Down
8 changes: 4 additions & 4 deletions Project.toml
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
name = "SpinGlassTensors"
uuid = "7584fc6a-5a23-4eeb-8277-827aab0146ea"
authors = ["Anna Maria Dziubyna <[email protected]>", "Tomasz Śmierzchalski <[email protected]>", "Bartłomiej Gardas <[email protected]>", "Konrad Jałowiecki <[email protected]>", "Łukasz Pawela <[email protected]>", "Marek M. Rams <[email protected]>"]
version = "1.1.3"
version = "1.2.0"

[deps]
CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
Expand All @@ -25,9 +25,9 @@ Memoization = "0.2.1"
NNlib = "0.9.13"
SparseArrays = "1.9"
TSVD = "0.4.4"
TensorOperations = "4"
cuTENSOR = "1.1.0"
julia = "1.9, 1.10"
TensorOperations = "5"
cuTENSOR = "2"
julia = "1.9, 1.10, 1.11"

[extras]
Logging = "56ddb016-857b-54e1-b83d-db4d58db5568"
Expand Down
1 change: 0 additions & 1 deletion src/SpinGlassTensors.jl
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,6 @@ using DocStringExtensions
using Base.Cartesian

import Base.Prehashed
# using SpinGlassNetworks

CUDA.allowscalar(false)

Expand Down
2 changes: 1 addition & 1 deletion src/contractions/dense.jl
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
const MatrixOrCuMatrix{R} = Union{
CuMatrix{R},
Matrix{R},
Diagonal{R,CuArray{R,1,Mem.DeviceBuffer}},
Diagonal{R,CuArray{R,1,CUDA.DeviceMemory}},
Diagonal{R,Vector{R}},
}

Expand Down
8 changes: 2 additions & 6 deletions src/mps/utils.jl
Original file line number Diff line number Diff line change
Expand Up @@ -43,9 +43,7 @@ end
function is_left_normalized::QMps, ::Val{true})
all(
eye(eltype(ψ), size(A, 2); toGPU = true) @tensor(
Id[x, y] := A[α, x, σ] * A[α, y, σ];
backend = cuTENSOR,
allocator = cuTENSOR
Id[x, y] := A[α, x, σ] * A[α, y, σ]
) for A values.tensors) # TODO: split the line
)
end
Expand All @@ -63,9 +61,7 @@ end
function is_right_normalized::QMps, ::Val{true})
all(
eye(eltype(ψ), size(B, 1); toGPU = true) @tensor(
Id[x, y] := B[x, α, σ] * B[y, α, σ];
backend = cuTENSOR,
allocator = cuTENSOR
Id[x, y] := B[x, α, σ] * B[y, α, σ]
) for B values.tensors) # TODO: split the line
)
end
Expand Down

0 comments on commit ff99e08

Please sign in to comment.