We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c4e8573 commit 94d1e00Copy full SHA for 94d1e00
src/nnpack/performance.jl
@@ -1,4 +1,16 @@
1
function select_threadpool(cdims::DenseConvDims, batch_size::Int)
2
+ inp_size = input_size(cdims)[1]
3
+ if batch_size >= 32
4
+ return shared_threadpool_dict[4][]
5
+ elseif batch_size >= 16 && inp_size >= 64
6
7
+ elseif inp_size <= 32
8
+ return C_NULL
9
+ elseif inp_size >= 128
10
11
+ elseif inp_size * batch_size >= 256
12
13
+ end
14
return C_NULL
15
end
16
0 commit comments