@@ -61,7 +61,7 @@ struct brgemm_cache_info_t {
6161};
6262
6363static std::vector<brgemm_cache_info_t > g_cache (DEFAULT_KERNEL_SIZE);
64- static int64_t kernel_id = -1 ;
64+ static int64_t g_kernel_id = -1 ;
6565
6666// TODO(haixin): use syscall to determine page size?
6767static constexpr size_t SCRATCH_SIZE = 2 * 4096 ;
@@ -81,39 +81,39 @@ int64_t dnnl_brgemm_dispatch(int64_t M, int64_t N, int64_t K, int64_t LDA,
8181 brgemm_strides_t stride_info{stride_a * dtypeA_size, stride_b * dtypeB_size};
8282
8383 write_lock_guard_t g (g_brgemm_lock);
84- kernel_id ++;
84+ g_kernel_id ++;
8585
86- if (kernel_id >= DEFAULT_KERNEL_SIZE) {
87- if (kernel_id >= (int64_t )g_cache.size ()) {
88- g_cache.resize (kernel_id + 1 );
86+ if (g_kernel_id >= DEFAULT_KERNEL_SIZE) {
87+ if (g_kernel_id >= (int64_t )g_cache.size ()) {
88+ g_cache.resize (g_kernel_id + 1 );
8989 }
9090 }
9191
9292 dnnl::impl::status_t status = brgemm_desc_init (
93- &g_cache[kernel_id ].desc , cpu_isa_t ::isa_undef,
93+ &g_cache[g_kernel_id ].desc , cpu_isa_t ::isa_undef,
9494 brgemm_batch_kind_t ::brgemm_strd, dnnl_dtypeA, dnnl_dtypeB,
9595 /* transA=*/ false , /* transB=*/ false , brgemm_layout_t ::brgemm_row_major,
9696 1 .0f , beta, LDA, LDB, LDC, M, N, K, &stride_info);
9797 assert (status == dnnl::impl::status::success &&
9898 " Failed to initialize BRGEMM descriptor" );
9999
100- status =
101- brgemm_kernel_create (&g_cache[kernel_id]. kernel , g_cache[kernel_id ].desc );
100+ status = brgemm_kernel_create (&g_cache[g_kernel_id]. kernel ,
101+ g_cache[g_kernel_id ].desc );
102102 assert (status == dnnl::impl::status::success &&
103103 " Failed to JIT BRGEMM kernel" );
104104
105105 brgemm_attr_t dnnl_attrs;
106- brgemm_desc_set_attr (&g_cache[kernel_id ].desc , dnnl_attrs);
106+ brgemm_desc_set_attr (&g_cache[g_kernel_id ].desc , dnnl_attrs);
107107
108- if (g_cache[kernel_id ].desc .is_tmm ) {
109- g_cache[kernel_id ].palette .reset (new char [PALETTE_SIZE]);
110- status = brgemm_init_tiles (g_cache[kernel_id ].desc ,
111- g_cache[kernel_id ].palette .get ());
108+ if (g_cache[g_kernel_id ].desc .is_tmm ) {
109+ g_cache[g_kernel_id ].palette .reset (new char [PALETTE_SIZE]);
110+ status = brgemm_init_tiles (g_cache[g_kernel_id ].desc ,
111+ g_cache[g_kernel_id ].palette .get ());
112112 assert (status == dnnl::impl::status::success &&
113113 " Failed to initialize palette for BRGEMM" );
114114 }
115115
116- return kernel_id ;
116+ return g_kernel_id ;
117117}
118118
119119void dnnl_brgemm_tileconfig (int64_t kernel_idx) {
0 commit comments