mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
Try to use avx_hs() - slow and requires alignment 4096 bits < (l.size*l.size*l.c)
May be faster only from 8192 bits and more.
This commit is contained in:
@ -684,6 +684,8 @@ void forward_convolutional_layer(convolutional_layer l, network_state state)
|
||||
// transpose B from NxK to KxN (x-axis (ldb = l.size*l.size*l.c) - should be multiple of 8 bits)
|
||||
{
|
||||
size_t ldb_align = 256;// 8;
|
||||
if (k > 4096)ldb_align = 4096;
|
||||
|
||||
size_t new_ldb = k + (ldb_align - k%ldb_align); // (k / 8 + 1) * 8;
|
||||
size_t t_intput_size = new_ldb * n;
|
||||
size_t t_bit_input_size = t_intput_size / 8;// +1;
|
||||
|
Reference in New Issue
Block a user