From b944e6b6dd04865a5e7f6591baf8988f6b0931b0 Mon Sep 17 00:00:00 2001 From: "Chereshnev, Eugene" Date: Wed, 18 Dec 2024 13:49:02 -0800 Subject: [PATCH] xe: conv: fix tuning support --- src/gpu/intel/jit/conv/config.cpp | 1 + src/gpu/intel/jit/conv/tiler.cpp | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/src/gpu/intel/jit/conv/config.cpp b/src/gpu/intel/jit/conv/config.cpp index d326f6bd172..9c9f6afa3e8 100644 --- 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 index ceff6baa377..1305aac899d 100644 --- 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;