Skip to content

Commit 0b99fbf

Browse files
committed
Removed Unrolled dependency
1 parent a247e6f commit 0b99fbf

File tree

3 files changed

+11
-12
lines changed

3 files changed

+11
-12
lines changed

Project.toml

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@ KernelAbstractions = "63c18a36-062a-441e-b654-da1e3ab1ce7c"
1010
Markdown = "d6f4376e-aef5-505a-96c1-9c027394607a"
1111
OhMyThreads = "67456a42-1dca-4109-a031-0a68de7e3ad5"
1212
Polyester = "f517fe37-dbe3-4b94-8317-1923a5111588"
13-
Unrolled = "9602ed7d-8fef-5bc8-8597-8f21381861e8"
1413

1514
[weakdeps]
1615
Metal = "dde4c033-4e86-420c-a63e-0dd931031962"
@@ -28,6 +27,5 @@ Markdown = "1"
2827
Metal = "1"
2928
OhMyThreads = "0.7, 0.8"
3029
Polyester = "0.7"
31-
Unrolled = "0.1"
3230
julia = "1.10"
3331
oneAPI = "1, 2"

src/AcceleratedKernels.jl

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@ using GPUArrays: GPUArrays, AbstractGPUVector, AbstractGPUArray, @allowscalar
1616
using KernelAbstractions
1717
using Polyester: @batch
1818
import OhMyThreads as OMT
19-
using Unrolled: @unroll, unrolled_map, FixedRange
2019

2120

2221
# Exposed functions from upstream packages

src/reduce/mapreduce_nd.jl

Lines changed: 11 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -327,16 +327,18 @@ function _mapreduce_nd_apply_init!(f, op, dst, src, backend, init, block_size)
327327
end
328328

329329

330-
function unrolled_map_index(f, tuple_vector::Tuple)
331-
unrolled_map(FixedRange{1, length(tuple_vector)}()) do i
332-
@inline f(i)
333-
end
330+
# Unrolled map constructing a tuple
331+
@inline function unrolled_map_index(f, tuple_vector::Tuple)
332+
_unrolled_map_index(f, tuple_vector, (), 1)
334333
end
335334

336335

337-
@inline @unroll function unrolled_foreach_index(f, tuple_vector::Tuple)
338-
@unroll for i in 1:length(tuple_vector)
339-
@inline f(tuple_vector, i)
340-
end
341-
nothing
336+
@inline function _unrolled_map_index(f, rest::Tuple{}, acc, i)
337+
acc
338+
end
339+
340+
341+
@inline function _unrolled_map_index(f, rest::Tuple, acc, i)
342+
result = f(i)
343+
_unrolled_map_index(f, Base.tail(rest), (acc..., result), i + 1)
342344
end

0 commit comments

Comments
 (0)