Skip to content

Commit d7a2b6b

Browse files
shubham.chaudharishubham.chaudhari
shubham.chaudhari
authored and
shubham.chaudhari
committed
Add thread throttling for dynamic arch neoversev1
1 parent a744897 commit d7a2b6b

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

interface/gemv.c

+6
Original file line numberDiff line numberDiff line change
@@ -127,6 +127,12 @@ static inline int get_gemv_optimal_nthreads(BLASLONG MN) {
127127
if (strcmp(gotoblas_corename(), "neoversev2") == 0) {
128128
return get_gemv_optimal_nthreads_neoversev2(MN, ncpu);
129129
}
130+
#elif defined(DYNAMIC_ARCH) && !defined(COMPLEX) && defined(DOUBLE) && !defined(BFLOAT16)
131+
if (strcmp(gotoblas_corename(), "neoversev1") == 0) {
132+
return get_dgemv_optimal_nthreads_neoversev1(MN, ncpu);
133+
}
134+
135+
130136
#endif
131137

132138
if ( MN < 115200L * GEMM_MULTITHREAD_THRESHOLD )

0 commit comments

Comments
 (0)