xe: conv: fix tuning support

oneapi-src · Dec 18, 2024 · b944e6b · b944e6b
1 parent 078f1d3
commit b944e6b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/gpu/intel/jit/conv/config.cpp b/src/gpu/intel/jit/conv/config.cpp
@@ -1794,6 +1794,7 @@ status_t init_cfg(conv_config_t &cfg, const primitive_t *prim) {
         auto try_cfg = cfg;
         auto status = try_init_cfg(try_cfg);
         if (status == status::success) {
+            if (cfg.tiler().is_tuning_mode()) cfg.tiler().move_next(cfg);
             cfg = std::move(try_cfg);
             return status::success;
         }

diff --git a/src/gpu/intel/jit/conv/tiler.cpp b/src/gpu/intel/jit/conv/tiler.cpp
@@ -1376,7 +1376,7 @@ class conv_tuner_t {
         if (tune_data_.reported_points() != created_configs_) return;
         if (created_configs_ < 8) return;
 
-        int beg = params_gen_.cur_index() + 1;
+        int beg = params_gen_.cur_index();
         int end = params_gen_.configs();
         if (beg == end) return;