From 25cf7517abf2095c625a908c1f3fa53eb41cfbf6 Mon Sep 17 00:00:00 2001
From: Arnav Sharma <arnav.sharma@amd.com>
Date: Thu, 23 Jun 2022 12:43:51 +0530
Subject: [PATCH] AOCL Dynamic Optimization for DGEMMT

- Optimized thread allocation for cases with n <= 220 for DGEMMT.

AMD-Internal: [CPUPL-2215]
Change-Id: Id01edf268a90fd96a41ef947db54f6afc490548f
---
 frame/base/bli_rntm.c | 54 +++++++++++++++++++++++++++++++++++++------
 1 file changed, 47 insertions(+), 7 deletions(-)

diff --git a/frame/base/bli_rntm.c b/frame/base/bli_rntm.c
index fbf5654b7..0db51870e 100644
--- a/frame/base/bli_rntm.c
+++ b/frame/base/bli_rntm.c
@@ -679,17 +679,57 @@ void bli_nthreads_optimum(
 	{
 		dim_t n = bli_obj_length(c);
 		dim_t k = bli_obj_width_after_trans(a);
-		dim_t product = (n*k)>>4; /* product is derived based on n and k */
 
-		//Limit the number thread for smaller sizes:
-		if(product <= 346)
+		if ( n < 32 )
 		{
-			n_threads_ideal = 1;
+			if ( k < 128 )
+			{
+				n_threads_ideal = 1;
+			}
+			else if ( k == 128 )
+			{
+				n_threads_ideal = 4;
+			}
 		}
-		/* finer threshold needs to set for max_thread cap of 2,3,4,5,6..32 */
-		else
+		else if ( n <= 40 )
 		{
-			n_threads_ideal = n_threads;
+			if ( k < 32 )
+			{
+				n_threads_ideal = 2;
+			}
+			else if ( k < 128 )
+			{
+				n_threads_ideal = 4;
+			}
+			else if ( k <= 256 )
+			{
+				n_threads_ideal = 8;
+			}
+		}
+		else if ( n < 115 )
+		{
+			if ( k < 128 )
+			{
+				n_threads_ideal = 6;
+			}
+			else if ( k <= 216 )
+			{
+				n_threads_ideal = 8;
+			}
+		}
+		else if ( n <= 160 )
+		{
+			if ( k <= 132 )
+			{
+				n_threads_ideal = 8;
+			}
+		}
+		else if ( n <= 220 )
+		{
+			if ( k < 128 )
+			{
+				n_threads_ideal = 8;
+			}
 		}
 	}
 	else if( family == BLIS_TRMM && bli_obj_is_double(c))