Browse Source

sbgemm: cooperlake: tuning for small matrix

tags/v0.3.18
Wangyang Guo 4 years ago
parent
commit
59a1114d03
1 changed files with 7 additions and 1 deletions
  1. +7
    -1
      kernel/x86_64/sbgemm_small_kernel_permit_cooperlake.c

+ 7
- 1
kernel/x86_64/sbgemm_small_kernel_permit_cooperlake.c View File

@@ -38,5 +38,11 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

int CNAME(int transa, int transb, BLASLONG M, BLASLONG N, BLASLONG K, FLOAT alpha, FLOAT beta)
{
return 1;
double MNK = (double) M * (double) N * (double) K;
if (MNK > 256.0*256.0*256.0) // disable for big size matrix
return 0;
/* small matrix kernel works well for N = 8, 16, 32 */
if (N == 8 || N == 16 || N == 32)
return 1;
return 0;
}

Loading…
Cancel
Save