• A
    Merge pull request #20658 from smbz:lstm_optimisation · ea7d4be3
    Andrew Ryrie 提交于
    * dnn: LSTM optimisation
    
    This uses the AVX-optimised fastGEMM1T for matrix multiplications where available, instead of the standard cv::gemm.
    
    fastGEMM1T is already used by the fully-connected layer.  This commit involves two minor modifications:
     - Use unaligned access.  I don't believe this involves any performance hit in on modern CPUs (Nehalem and Bulldozer onwards) in the case where the address is actually aligned.
     - Allow for weight matrices where the number of columns is not a multiple of 8.
    
    I have not enabled AVX-512 as I don't have an AVX-512 CPU to test on.
    
    * Fix warning about initialisation order
    
    * Remove C++11 syntax
    
    * Fix build when AVX(2) is not available
    
    In this case the CV_TRY_X macros are defined to 0, rather than being undefined.
    
    * Minor changes as requested:
    
     - Don't check hardware support for AVX(2) when dispatch is disabled for these
     - Add braces
    
    * Fix out-of-bounds access in fully connected layer
    
    The old tail handling in fastGEMM1T implicitly rounded vecsize up to the next multiple of 8, and the fully connected layer implements padding up to the next multiple of 8 to cope with this.  The new tail handling does not round the vecsize upwards like this but it does require that the vecsize is at least 8.  To adapt to the new tail handling, the fully connected layer now rounds vecsize itself at the same time as adding the padding(which makes more sense anyway).
    
    This also means that the fully connected layer always passes a vecsize of at least 8 to fastGEMM1T, which fixes the out-of-bounds access problems.
    
    * Improve tail mask handling
    
     - Use static array for generating tail masks (as requested)
     - Apply tail mask to the weights as well as the input vectors to prevent spurious propagation of NaNs/Infs
    
    * Revert whitespace change
    
    * Improve readability of conditions for using AVX
    
    * dnn(lstm): minor coding style changes, replaced left aligned load
    ea7d4be3
fully_connected_layer.cpp 22.4 KB