Skip to content

Commit

Permalink
Update default M=8, so when matrix M<=8, we will use ig kernel to boo…
Browse files Browse the repository at this point in the history
…st bf16 gemms, otherwise use oneDNN kernel.
  • Loading branch information
a3213105 committed Nov 1, 2023
1 parent d8838af commit 3f67cc7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/utils/matmul_helper.h
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@
#include <map>
#include <tuple>

#define USE_AMX_M 2
#define USE_AMX_M 8

class MMHelper {
public:
Expand Down

0 comments on commit 3f67cc7

Please sign in to comment.