frozenleaves/oneDNN
1
0
Fork 0
mirror of https://github.com/uxlfoundation/oneDNN.git synced 2025-10-20 10:03:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

intel: rnn: disable larger cell sizes for fusion in systolic arch

Browse Source
This commit is contained in:
Sadia, Haleema
2025-10-03 10:33:52 -07:00
committed by Haleema Sadia
parent 11c30994ae
commit b0d9796c0f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/gpu/intel/rnn/utils.cpp
View File

@ -185,7 +185,7 @@ void init_conf(conf_t &conf, const desc_t &rd,
if (rd.cell_kind == alg_kind::vanilla_lstm) {
min_k = (min_k <= 256) ? 160 : 256;
} else {
min_k = 256;
min_k = device_info.mayiuse_systolic() ? 64 : 256;
}
dim_t k_limit = tail_dhc ? 50 : nstl::min(min_k, ideal_k);