Skip to content

Commit 3646d91

Browse files
committed
xe: conv: enable selecting strategy by desc index
1 parent 4ebaabc commit 3646d91

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

src/gpu/intel/conv/jit.cpp

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -145,9 +145,15 @@ class gen_t {
145145
tiler->set_cur_version(primitive->version());
146146
}
147147

148+
int desc_idx = gpu_utils::dev_getenv("desc_idx", -1);
148149
for (int try_iter = 0; try_iter < max_tries; try_iter++) {
149-
if (try_iter != 0 && !tiler->is_tuning_mode())
150+
if (try_iter < desc_idx) {
150151
tiler->move_next(cfg);
152+
continue;
153+
}
154+
if ((try_iter != 0 && !tiler->is_tuning_mode())) {
155+
tiler->move_next(cfg);
156+
}
151157
try {
152158
cfg = data.pd_cfg;
153159
cfg.set_pd(pd);

0 commit comments

Comments
 (0)