diff --git a/ext/TensorOperationsGPUArraysExt.jl b/ext/TensorOperationsGPUArraysExt.jl index 3bca5852..e665180f 100644 --- a/ext/TensorOperationsGPUArraysExt.jl +++ b/ext/TensorOperationsGPUArraysExt.jl @@ -20,4 +20,9 @@ function TO.unsafe_buffer_wrap( return GPUArrays.derive(T, buffer.buffer, TO._asdims(structure), Int(start) รท sizeof(T)) end +function TO.tensorfree!(C::AnyGPUArray, ::TO.DefaultAllocator) + GPUArrays.unsafe_free!(C) + return nothing +end + end