loongarch64: Add buffer offset for target LOONGSON3R5

XiWeiGu · May 8, 2024 · 21dd649 · 21dd649
1 parent 6017ad7
commit 21dd649
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 15 deletions.
diff --git a/driver/others/blas_server.c b/driver/others/blas_server.c
@@ -1074,13 +1074,10 @@ fprintf(STDERR, "Server[%2ld] Calculation started.  Mode = 0x%03x M = %3ld N=%3l
       main_status[cpu] = MAIN_RUNNING1;
 #endif
 
-//For Loongson servers, like the 3C5000 (featuring 16 cores), applying an
-//offset to the buffer is essential for minimizing cache conflicts and optimizing performance.
+//For target LOONGSON3R5, applying an offset to the buffer is essential
+//for minimizing cache conflicts and optimizing performance.
 #if defined(LOONGSON3R5) && !defined(NO_AFFINITY)
-      char model_name[128];
-      get_cpu_model(model_name);
-      if ((strstr(model_name, "3C5000") != NULL) || (strstr(model_name, "3D5000") != NULL))
-        if (sa == NULL) sa = (void *)((BLASLONG)buffer + (WhereAmI() & 0xf) * GEMM_OFFSET_A);
+      if (sa == NULL) sa = (void *)((BLASLONG)buffer + (WhereAmI() & 0xf) * GEMM_OFFSET_A);
 #endif
       if (sa == NULL) sa = (void *)((BLASLONG)buffer + GEMM_OFFSET_A);
 
@@ -1157,4 +1154,4 @@ if (queue -> mode & BLAS_PTHREAD) {
 
 }
 
-#endif
+#endif
diff --git a/interface/gemm.c b/interface/gemm.c
@@ -521,15 +521,10 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE TransA, enum CBLAS_TRANS
 
   buffer = (XFLOAT *)blas_memory_alloc(0);
 
-//For Loongson servers, like the 3C5000 (featuring 16 cores), applying an
-//offset to the buffer is essential for minimizing cache conflicts and optimizing performance.
+//For target LOONGSON3R5, applying an offset to the buffer is essential
+//for minimizing cache conflicts and optimizing performance.
 #if defined(LOONGSON3R5) && !defined(NO_AFFINITY)
-  char model_name[128];
-  get_cpu_model(model_name);
-  if ((strstr(model_name, "3C5000") != NULL) || (strstr(model_name, "3D5000") != NULL))
-    sa = (XFLOAT *)((BLASLONG)buffer + (WhereAmI() & 0xf) * GEMM_OFFSET_A);
-  else
-    sa = (XFLOAT *)((BLASLONG)buffer + GEMM_OFFSET_A);
+  sa = (XFLOAT *)((BLASLONG)buffer + (WhereAmI() & 0xf) * GEMM_OFFSET_A);
 #else
   sa = (XFLOAT *)((BLASLONG)buffer +GEMM_OFFSET_A);
 #endif