From e1374ea221b95f2ac61ee95feda5fad6896b3a27 Mon Sep 17 00:00:00 2001
From: apoorva <apoorva@streamhpc.com>
Date: Wed, 9 Jul 2025 08:25:30 +0000
Subject: [PATCH] Revert "REVERTED THE PROFILER CHANGES"

This reverts commit 21cb98546cca10c80fea142544c3f9d0937614f0.
---
 profiler/include/profiler/profile_gemm_add_relu_impl.hpp | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/profiler/include/profiler/profile_gemm_add_relu_impl.hpp b/profiler/include/profiler/profile_gemm_add_relu_impl.hpp
index 8b0d4cd79d..65b1925f61 100644
--- a/profiler/include/profiler/profile_gemm_add_relu_impl.hpp
+++ b/profiler/include/profiler/profile_gemm_add_relu_impl.hpp
@@ -1,5 +1,5 @@
 // SPDX-License-Identifier: MIT
-// Copyright (c) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
+// Copyright (c) 2025, Advanced Micro Devices, Inc. All rights reserved.
 
 #pragma once
 
@@ -93,7 +93,7 @@ bool profile_gemm_add_relu_impl(int do_verification,
     const auto b_element_op   = BElementOp{};
     const auto cde_element_op = CDEElementOp{};
 
-    using DeviceOp = ck::tensor_operation::device::DeviceGemmMultipleD<
+    using DeviceOp = ck::tensor_operation::device::DeviceGemmMultipleDSplitK<
         ALayout,
         BLayout,
         ck::Tuple<D0Layout>,
@@ -173,6 +173,7 @@ bool profile_gemm_add_relu_impl(int do_verification,
             StrideB,
             std::array<ck::index_t, 1>{StrideD0},
             StrideE,
+            1,
             a_element_op,
             b_element_op,
             cde_element_op);