Skip to content

[WIP,Testing] Make GEMM_DIVIDE_LIMIT available to DYNAMIC_ARCH builds #5688

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
martin-frbg wants to merge 3 commits into
base: develop
Choose a base branch
from
Open

[WIP,Testing] Make GEMM_DIVIDE_LIMIT available to DYNAMIC_ARCH builds #5688

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
d7b13fe
Provide a default GEMM_DIVIDE_LIMIT and add it to DYNAMIC_ARCH
martin-frbg Mar 19, 2026
8f5e495
Add GEMM_DIVIDE_LIMIT to parameters
martin-frbg Mar 19, 2026
b7601ea
Retrieve cpu-specific GEMM_DIVIDE_LIMIT if DYNAMIC_ARCH
martin-frbg Mar 19, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions common_param.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -47,6 +47,7 @@
typedef struct {
int dtb_entries;
int switch_ratio;
int divide_limit;
int offsetA, offsetB, align;
#if BUILD_HFLOAT16 == 1
int shgemm_p, shgemm_q, shgemm_r;
Expand Down
4 changes: 4 additions & 0 deletions driver/level3/gemm.c
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -63,6 +63,10 @@
#define DIVIDE_RATE GEMM_DIVIDE_RATE
#endif

#ifdef DYNAMIC_ARCH
#define GEMM_DIVIDE_LIMIT gotoblas->divide_limit
#endif

#ifdef GEMM_DIVIDE_LIMIT
#define DIVIDE_LIMIT GEMM_DIVIDE_LIMIT
#endif
Expand Down
2 changes: 2 additions & 0 deletions kernel/setparam-ref.c
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -54,6 +54,8 @@ gotoblas_t TABLE_NAME = {

SWITCH_RATIO,

GEMM_DIVIDE_LIMIT,

GEMM_DEFAULT_OFFSET_A, GEMM_DEFAULT_OFFSET_B, GEMM_DEFAULT_ALIGN,

#ifdef BUILD_HFLOAT16
Expand Down
4 changes: 4 additions & 0 deletions param.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4260,6 +4260,10 @@ Until then, just keep it different than DGEMM_DEFAULT_UNROLL_N to keep copy rout
#define SWITCH_RATIO 2
#endif

#ifndef GEMM_DIVIDE_LIMIT
#define GEMM_DIVIDE_LIMIT 0
#endif

#ifndef QGEMM_DEFAULT_UNROLL_M
#define QGEMM_DEFAULT_UNROLL_M 2
#endif
Expand Down
Loading