From cd13e01bad22948d86189117c3e6cff487bfa2e5 Mon Sep 17 00:00:00 2001
From: AntonOresten <antonoresten@proton.me>
Date: Sat, 27 Jun 2026 17:03:24 +0200
Subject: [PATCH 1/3] Add cuTileExt

---
 Project.toml     |  9 +++++++++
 ext/cuTileExt.jl | 42 ++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 51 insertions(+)
 create mode 100644 ext/cuTileExt.jl

diff --git a/Project.toml b/Project.toml
index 6646343..c8cb451 100644
--- a/Project.toml
+++ b/Project.toml
@@ -3,11 +3,20 @@ uuid = "b58c8408-13c4-4787-8733-7038ae624acf"
 version = "0.2.1"
 authors = ["Anton Oresten <antonoresten@proton.me> and contributors"]
 
+[workspace]
+projects = ["test", "docs"]
+
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
 Republic = "27243419-9dde-4721-b67c-fd63626fea7f"
 StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
 
+[weakdeps]
+cuTile = "0dea8319-8c4a-4662-a73d-20234d115b9a"
+
+[extensions]
+cuTileExt = "cuTile"
+
 [compat]
 Adapt = "4.6.1"
 Republic = "2.2.0"
diff --git a/ext/cuTileExt.jl b/ext/cuTileExt.jl
new file mode 100644
index 0000000..2b1e64c
--- /dev/null
+++ b/ext/cuTileExt.jl
@@ -0,0 +1,42 @@
+module cuTileExt
+
+using BitPacking: NarrowArray, bitwidth
+
+import Adapt
+import cuTile as ct
+using cuTile: KernelAdaptor, TileArray
+
+struct ReinterpretTileArray{T,N,A<:TileArray{UInt8,N}}
+    parent::A
+end
+
+Base.parent(arr::ReinterpretTileArray) = arr.parent
+Base.eltype(::ReinterpretTileArray{T}) where T = T
+Base.ndims(::ReinterpretTileArray{T,N}) where {T,N} = N
+
+function Base.size(arr::ReinterpretTileArray, i::Integer)
+    ratio = 8 ÷ bitwidth(eltype(arr))
+    return i == 1 ? size(parent(arr), i) * ratio : size(parent(arr), i)
+end
+Base.size(arr::ReinterpretTileArray) = ntuple(i -> size(arr, i), Val(ndims(arr)))
+
+function Adapt.adapt_structure(to::KernelAdaptor, arr::NarrowArray)
+    parent = Adapt.adapt(to, reinterpret(UInt8, arr))
+    return ReinterpretTileArray{eltype(arr),ndims(parent),typeof(parent)}(parent)
+end
+
+function ct.store(arr::ReinterpretTileArray, index, tile; kws...)
+    return ct.store(parent(arr), index, reinterpret(UInt8, tile); kws...)
+end
+
+function ct.load(arr::ReinterpretTileArray, index, shape; kws...)
+    ratio = 8 ÷ bitwidth(eltype(arr))
+    shape′ = ntuple(Val(ndims(arr))) do i
+        i == 1 ? shape[i] ÷ ratio : shape[i]
+    end
+    byte_tile = ct.load(parent(arr), index, shape′; kws...)
+    tile = reinterpret(eltype(arr), byte_tile)
+    return tile
+end
+
+end

From 04dabe01bda39f0d6de23d4ecca9e08198cc30aa Mon Sep 17 00:00:00 2001
From: AntonOresten <antonoresten@proton.me>
Date: Sat, 27 Jun 2026 20:27:35 +0200
Subject: [PATCH 2/3] Use AbstractTileArray

---
 ext/cuTileExt.jl | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/ext/cuTileExt.jl b/ext/cuTileExt.jl
index 2b1e64c..d3acc67 100644
--- a/ext/cuTileExt.jl
+++ b/ext/cuTileExt.jl
@@ -4,9 +4,8 @@ using BitPacking: NarrowArray, bitwidth
 
 import Adapt
 import cuTile as ct
-using cuTile: KernelAdaptor, TileArray
 
-struct ReinterpretTileArray{T,N,A<:TileArray{UInt8,N}}
+struct ReinterpretTileArray{T,N,A<:ct.TileArray{UInt8,N}} <: ct.AbstractTileArray{T,N}
     parent::A
 end
 
@@ -20,7 +19,7 @@ function Base.size(arr::ReinterpretTileArray, i::Integer)
 end
 Base.size(arr::ReinterpretTileArray) = ntuple(i -> size(arr, i), Val(ndims(arr)))
 
-function Adapt.adapt_structure(to::KernelAdaptor, arr::NarrowArray)
+function Adapt.adapt_structure(to::ct.KernelAdaptor, arr::NarrowArray)
     parent = Adapt.adapt(to, reinterpret(UInt8, arr))
     return ReinterpretTileArray{eltype(arr),ndims(parent),typeof(parent)}(parent)
 end

From 4bf4e20249a53870ec1881dbc338ddaf80ec5bfb Mon Sep 17 00:00:00 2001
From: AntonOresten <antonoresten@proton.me>
Date: Wed, 1 Jul 2026 09:29:27 +0200
Subject: [PATCH 3/3] Update test/Project.toml

---
 test/Project.toml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/test/Project.toml b/test/Project.toml
index 0c36332..9597d47 100644
--- a/test/Project.toml
+++ b/test/Project.toml
@@ -1,2 +1,3 @@
 [deps]
+BitPacking = "b58c8408-13c4-4787-8733-7038ae624acf"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"