Invoke the syscall to allow the use of amx tiles

2023-05-19 00:21:16 +08:00
parent 0b83088887
commit 90f041e348
3 changed files with 39 additions and 0 deletions
--- a/common_param.h
+++ b/common_param.h
@@ -53,6 +53,7 @@ typedef struct {
  int sbgemm_p, sbgemm_q, sbgemm_r;
  int sbgemm_unroll_m, sbgemm_unroll_n, sbgemm_unroll_mn;
  int sbgemm_align_k;
+  int need_amxtile_permission;  // 0 default, 1 for device support amx.

  void   (*sbstobf16_k) (BLASLONG, float    *, BLASLONG, bfloat16 *, BLASLONG);
  void   (*sbdtobf16_k) (BLASLONG, double   *, BLASLONG, bfloat16 *, BLASLONG);