Revert "REVERTED THE PROFILER CHANGES"

This reverts commit 21cb98546c.
This commit is contained in:
apoorva
2025-07-09 08:25:30 +00:00
parent 21cb98546c
commit e1374ea221

View File

@@ -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);