Skip to content

Commit 8b776a7

Browse files
authored
Update to GPUArrays 0.9. (#2068)
* Switch to GPUArrays buffer management. * Support mixed-type constructors.
1 parent 95cd0d9 commit 8b776a7

File tree

9 files changed

+98
-349
lines changed

9 files changed

+98
-349
lines changed

Manifest.toml

+3-3
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
julia_version = "1.8.5"
44
manifest_format = "2.0"
5-
project_hash = "e9075ab949829f915a85537ba051d4224e4d2f06"
5+
project_hash = "668ed29932abfaac250cb6c23a1bfa769a4b4fcf"
66

77
[[deps.AbstractFFTs]]
88
deps = ["ChainRulesCore", "LinearAlgebra", "Test"]
@@ -146,9 +146,9 @@ uuid = "9fa8497b-333b-5362-9e8d-4d0656e87820"
146146

147147
[[deps.GPUArrays]]
148148
deps = ["Adapt", "GPUArraysCore", "LLVM", "LinearAlgebra", "Printf", "Random", "Reexport", "Serialization", "Statistics"]
149-
git-tree-sha1 = "2e57b4a4f9cc15e85a24d603256fe08e527f48d1"
149+
git-tree-sha1 = "8ad8f375ae365aa1eb2f42e2565a40b55a4b69a8"
150150
uuid = "0c68f7d7-f131-5f86-a1c3-88cf8149b2d7"
151-
version = "8.8.1"
151+
version = "9.0.0"
152152

153153
[[deps.GPUArraysCore]]
154154
deps = ["Adapt"]

Project.toml

+1-1
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ CUDA_Runtime_jll = "0.9"
5151
Crayons = "4"
5252
DataFrames = "1"
5353
ExprTools = "0.1"
54-
GPUArrays = "8.6"
54+
GPUArrays = "9"
5555
GPUCompiler = "0.23"
5656
KernelAbstractions = "0.9.2"
5757
LLVM = "6"

lib/cutensor/src/wrappers.jl

+2-5
Original file line numberDiff line numberDiff line change
@@ -27,11 +27,8 @@ end
2727

2828
abstract type CuTensorPlan end
2929

30-
function CUDA.unsafe_free!(plan::CuTensorPlan, stream::CuStream=stream())
31-
CUDA.unsafe_free!(plan.workspace, stream)
32-
end
33-
34-
unsafe_finalize!(plan::CuTensorPlan) = CUDA.unsafe_free!(plan, default_stream())
30+
CUDA.unsafe_free!(plan::CuTensorPlan) = CUDA.unsafe_free!(plan.workspace)
31+
unsafe_finalize!(plan::CuTensorPlan) = CUDA.unsafe_finalize!(plan.workspace)
3532

3633

3734
const ModeType = AbstractVector{<:Union{Char, Integer}}

0 commit comments

Comments
 (0)