Skip to content

Commit

Permalink
upgrade CUDA (#153)
Browse files Browse the repository at this point in the history
  • Loading branch information
GiggleLiu authored Feb 17, 2023
1 parent 556df4f commit 73e7c27
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
4 changes: 2 additions & 2 deletions Project.toml
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
name = "OMEinsum"
uuid = "ebe7aa44-baf0-506c-a96f-8464559b3922"
authors = ["Andreas Peter <[email protected]>"]
version = "0.7.2"
version = "0.7.3"

[deps]
AbstractTrees = "1520ce14-60c1-5f80-bbc7-55ef81b5835c"
Expand All @@ -19,7 +19,7 @@ TupleTools = "9d95972d-f1c8-5527-a6e0-b4b365fa01f6"
[compat]
AbstractTrees = "0.3, 0.4"
BatchedRoutines = "0.2"
CUDA = "3.10"
CUDA = "4"
ChainRulesCore = "1"
Combinatorics = "1.0"
MacroTools = "0.5"
Expand Down
4 changes: 2 additions & 2 deletions src/cueinsum.jl
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ _unwrap(x::LinearAlgebra.Adjoint{T,<:CuArray{T}}) where T = CuArray(x)
_unwrap(x::LinearAlgebra.Transpose{T,<:CuArray{T}}) where T = CuArray(x)
_unwrap(x::CuArray) = x

asarray(x, arr::CuArray) where T = CuArray(fill(x, ()))
asarray(x, arr::CuArray) = CuArray(fill(x, ()))
asarray(x::AbstractArray, y::CuArray) = x
asscalar(x::DenseCuArray) = Array(x)[]

Expand Down Expand Up @@ -64,7 +64,7 @@ end
Expr(:tuple, ids...)
end

function expanddims!(::Val{ixs}, ::Val{iy}, x, y) where {LT,ixs,iy}
function expanddims!(::Val{ixs}, ::Val{iy}, x, y) where {ixs,iy}
nthreads = 256
nblocks = cld(prod(size(x)), nthreads)
CIS = CartesianIndices(x)
Expand Down

0 comments on commit 73e7c27

Please sign in to comment.