From 4e32dbd74f54a2d1e163925cf29c1209a96acfef Mon Sep 17 00:00:00 2001
From: Katharine Hyatt <kslimes@gmail.com>
Date: Thu, 20 Aug 2026 14:39:09 +0200
Subject: [PATCH] Add a tensorfree! for GPUArray + DefaultAllocator

---
 ext/TensorOperationsGPUArraysExt.jl | 5 +++++
 1 file changed, 5 insertions(+)

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