"magic_pdf/vscode:/vscode.git/clone" did not exist on "779d2e8aaf6dcb62e7211bbd2cfd8554dd80ffe8"
Commit a74b2263 authored by Adam Osewski's avatar Adam Osewski
Browse files

Clean up.

parent 522b7aee
// SPDX-License-Identifier: MIT
// Copyright (c) 2023, Advanced Micro Devices, Inc. All rights reserved.
// Copyright (c) 2024, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
......@@ -75,12 +75,11 @@ class StridedReductionTileLoop
///
/// @return The workgroup flag index.
///
__device__ uint32_t GetWorkgroupFlagIdx([[maybe_unused]] index_t k_tiles,
__device__ uint32_t GetWorkgroupFlagIdx(index_t k_tiles,
index_t output_tile_idx,
index_t output_tile_idx_offset) const
{
return (output_tile_idx + output_tile_idx_offset) % GetFlagCount(k_tiles);
// return output_tile_idx + output_tile_idx_offset;
}
///
......
......@@ -307,12 +307,12 @@ bool profile_ggemm_multid_splitk(int do_verification,
<< (instance_pass ? "SUCCEED" : "FAILED") << std::endl;
pass = pass && instance_pass;
std::cout << ">>>>>CPU verification end!" << std::endl;
// std::cout << ">>>>>CPU verification end!" << std::endl;
}
if(time_kernel)
{
std::cout << ">>>>>GPU time profiling start!" << std::endl;
// std::cout << ">>>>>GPU time profiling start!" << std::endl;
float avg_time = invoker_ptr->Run(
argument_ptr.get(),
StreamConfig{nullptr, time_kernel, 0, warmup_iter, kernel_iter});
......
......@@ -91,9 +91,9 @@ int profile_grouped_gemm_multiple_d_splitk(int argc, char* argv[])
const auto StrideAs = argToIntArray(argv[11]);
const auto StrideBs = argToIntArray(argv[12]);
const auto StrideCs = argToIntArray(argv[13]);
const int kbatch = argc == 15 ? std::stoi(argv[14]) : 1;
const int warmup_iter = argc == 16 ? std::stoi(argv[15]) : 1;
const int kernel_iter = argc == 17 ? std::stoi(argv[16]) : 10;
const int kbatch = argc >= 15 ? std::stoi(argv[14]) : 1;
const int warmup_iter = argc >= 16 ? std::stoi(argv[15]) : 1;
const int kernel_iter = argc >= 17 ? std::stoi(argv[16]) : 10;
#ifdef CK_ENABLE_FP16
if(data_type == GemmDataType::F16_F16_F16 && layout == GemmMatrixLayout::MK_KN_MN)
{
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment