Update to GPUArrays 0.9. (#2068)

* Switch to GPUArrays buffer management. * Support mixed-type constructors.
JuliaGPU · Sep 1, 2023 · 8b776a7 · 8b776a7
1 parent 95cd0d9
commit 8b776a7
Show file tree

Hide file tree

Showing 9 changed files with 98 additions and 349 deletions.
diff --git a/Manifest.toml b/Manifest.toml
@@ -2,7 +2,7 @@
 
 julia_version = "1.8.5"
 manifest_format = "2.0"
-project_hash = "e9075ab949829f915a85537ba051d4224e4d2f06"
+project_hash = "668ed29932abfaac250cb6c23a1bfa769a4b4fcf"
 
 [[deps.AbstractFFTs]]
 deps = ["ChainRulesCore", "LinearAlgebra", "Test"]
@@ -146,9 +146,9 @@ uuid = "9fa8497b-333b-5362-9e8d-4d0656e87820"
 
 [[deps.GPUArrays]]
 deps = ["Adapt", "GPUArraysCore", "LLVM", "LinearAlgebra", "Printf", "Random", "Reexport", "Serialization", "Statistics"]
-git-tree-sha1 = "2e57b4a4f9cc15e85a24d603256fe08e527f48d1"
+git-tree-sha1 = "8ad8f375ae365aa1eb2f42e2565a40b55a4b69a8"
 uuid = "0c68f7d7-f131-5f86-a1c3-88cf8149b2d7"
-version = "8.8.1"
+version = "9.0.0"
 
 [[deps.GPUArraysCore]]
 deps = ["Adapt"]

diff --git a/Project.toml b/Project.toml
@@ -51,7 +51,7 @@ CUDA_Runtime_jll = "0.9"
 Crayons = "4"
 DataFrames = "1"
 ExprTools = "0.1"
-GPUArrays = "8.6"
+GPUArrays = "9"
 GPUCompiler = "0.23"
 KernelAbstractions = "0.9.2"
 LLVM = "6"

diff --git a/lib/cutensor/src/wrappers.jl b/lib/cutensor/src/wrappers.jl
@@ -27,11 +27,8 @@ end
 
 abstract type CuTensorPlan end
 
-function CUDA.unsafe_free!(plan::CuTensorPlan, stream::CuStream=stream())
-    CUDA.unsafe_free!(plan.workspace, stream)
-end
-
-unsafe_finalize!(plan::CuTensorPlan) = CUDA.unsafe_free!(plan, default_stream())
+CUDA.unsafe_free!(plan::CuTensorPlan) = CUDA.unsafe_free!(plan.workspace)
+unsafe_finalize!(plan::CuTensorPlan) = CUDA.unsafe_finalize!(plan.workspace)
 
 
 const ModeType = AbstractVector{<:Union{Char, Integer}}