Skip to content

Commit 36b4d9b

Browse files
committed
Don't measure pooling directly, due to inference limitations
1 parent 36d50de commit 36b4d9b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/impl/pooling_direct.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ using Statistics
33
# Pooling is so similar, we abstract over meanpooling and maxpooling, simply replacing
44
# the inner loop operation and a few initialization parameters.
55
for name in (:max, :mean)
6-
@eval @timeit_debug to function $((Symbol("$(name)pool_direct!")))(
6+
@eval function $((Symbol("$(name)pool_direct!")))(
77
y::AbstractArray{T,5}, x::AbstractArray{T,5},
88
pdims::PoolDims; alpha::T = T(1), beta::T = T(0)) where {T}
99
check_dims(size(x), size(y), pdims)
@@ -121,7 +121,7 @@ for name in (:max, :mean)
121121

122122
# Same story for gradients, and although this is very similar to the forward pass,
123123
# it's unfortunately different enough that I think we need a separate function. :(
124-
@eval @timeit_debug to function $((Symbol("$(name)pool_direct!")))(
124+
@eval function $((Symbol("$(name)pool_direct!")))(
125125
dx::AbstractArray{T,5}, dy::AbstractArray{T,5},
126126
y::AbstractArray{T,5}, x::AbstractArray{T,5},
127127
pdims::PoolDims; alpha::T = T(1), beta::T = T(0)) where {T}

0 commit comments

Comments
 (0)