diff --git a/Project.toml b/Project.toml index b60290e..8c0bb7a 100644 --- a/Project.toml +++ b/Project.toml @@ -1,10 +1,6 @@ name = "TensorOperations" uuid = "6aa20fa7-93e2-5fca-9bc0-fbd0db3c71a2" -authors = [ - "Lukas Devos ", - "Maarten Van Damme ", - "Jutho Haegeman ", -] +authors = ["Lukas Devos ", "Maarten Van Damme ", "Jutho Haegeman "] version = "5.0.2" [deps] @@ -33,7 +29,7 @@ TensorOperationscuTENSORExt = ["cuTENSOR", "CUDA"] [compat] Aqua = "0.6, 0.7, 0.8" -Bumper = "0.6" +Bumper = "0.6, 0.7" CUDA = "5.4.0" ChainRulesCore = "1" ChainRulesTestUtils = "1" @@ -64,14 +60,4 @@ Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40" cuTENSOR = "011b41b2-24ef-40a8-b3eb-fa098493e9e1" [targets] -test = [ - "Test", - "Random", - "DynamicPolynomials", - "ChainRulesTestUtils", - "CUDA", - "cuTENSOR", - "Aqua", - "Logging", - "Bumper", -] +test = ["Test", "Random", "DynamicPolynomials", "ChainRulesTestUtils", "CUDA", "cuTENSOR", "Aqua", "Logging", "Bumper"] diff --git a/ext/TensorOperationsBumperExt.jl b/ext/TensorOperationsBumperExt.jl index 44c3719..0b0a1ac 100644 --- a/ext/TensorOperationsBumperExt.jl +++ b/ext/TensorOperationsBumperExt.jl @@ -7,7 +7,7 @@ function TensorOperations.tensoralloc(::Type{A}, structure, ::Val{istemp}, buf::Union{SlabBuffer,AllocBuffer}) where {A<:AbstractArray, istemp} # TODO: remove the `ndims` check if this is fixed in Bumper / StrideArraysCore - if istemp & ndims(A) > 0 + if istemp && ndims(A) > 0 return Bumper.alloc!(buf, eltype(A), structure...) else return TensorOperations.tensoralloc(A, structure, Val(istemp))