From a34dfa1e4e7fa9283c04ba0bce262bb7808105d5 Mon Sep 17 00:00:00 2001 From: Uros Petkovic Date: Thu, 6 Aug 2026 14:55:19 +0200 Subject: [PATCH] Adding mlss by default for gfx115x --- src/migraphx/mgx_ep.cc | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/src/migraphx/mgx_ep.cc b/src/migraphx/mgx_ep.cc index c5442c2..6659153 100644 --- a/src/migraphx/mgx_ep.cc +++ b/src/migraphx/mgx_ep.cc @@ -492,19 +492,19 @@ ExecutionProvider::ExecutionProvider(const ProviderFactory& factory, std::string PARSE_ENV_VAR(env_var::kMlssUseSpecificOps, mlss_use_specific_ops_); PARSE_ENV_VAR(env_var::kModelArch, model_arch_); - // Per-architecture ops to force onto AMDMLSS. - // Add a row here to enable specific ops on additional architectures. + // Per-architecture-family ops to force onto AMDMLSS. + // Prefixes also match architecture variants and HIP feature suffixes. struct arch_mlss_ops { - std::string_view arch; + std::string_view arch_prefix; std::string_view ops; // comma-separated op names }; static constexpr std::array kArchMlssOps{{ - {"gfx1200", "conv"}, - {"gfx1201", "conv"}, + {"gfx12", "conv"}, + {"gfx115", "conv"}, }}; - for (const auto& [arch, ops] : kArchMlssOps) { - if (compute_capability_.rfind(arch, 0) == 0) { + for (const auto& [arch_prefix, ops] : kArchMlssOps) { + if (compute_capability_.rfind(arch_prefix, 0) == 0) { if (!mlss_use_specific_ops_.empty()) { mlss_use_specific_ops_ += ","; }