Skip to content

Commit 6e92528

Browse files
author
ZhangYan
committed
fix comment
1 parent 5c04e70 commit 6e92528

File tree

1 file changed

+14
-14
lines changed

1 file changed

+14
-14
lines changed

lib/gc/ExecutionEngine/CPURuntime/Microkernel/BrgemmOnednn.cpp

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ struct brgemm_cache_info_t {
6161
};
6262

6363
static 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?
6767
static 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

119119
void dnnl_brgemm_tileconfig(int64_t kernel_idx) {

0 commit comments

Comments
 (0)