Browse Source

Add thread throttling for dynamic arch neoversev1

tags/v0.3.30
shubham.chaudhari 7 months ago
parent
commit
189dbbc04f
1 changed files with 6 additions and 0 deletions
  1. +6
    -0
      interface/gemv.c

+ 6
- 0
interface/gemv.c View File

@@ -127,6 +127,12 @@ static inline int get_gemv_optimal_nthreads(BLASLONG MN) {
if (strcmp(gotoblas_corename(), "neoversev2") == 0) {
return get_gemv_optimal_nthreads_neoversev2(MN, ncpu);
}
#elif defined(DYNAMIC_ARCH) && !defined(COMPLEX) && defined(DOUBLE) && !defined(BFLOAT16)
if (strcmp(gotoblas_corename(), "neoversev1") == 0) {
return get_dgemv_optimal_nthreads_neoversev1(MN, ncpu);
}


#endif

if ( MN < 115200L * GEMM_MULTITHREAD_THRESHOLD )


Loading…
Cancel
Save