split_k_arg.hpp Source File

split_k_arg.hpp Source File#

Composable Kernel: split_k_arg.hpp Source File

// SPDX-License-Identifier: MIT
// Copyright (c) 2025, Advanced Micro Devices, Inc. All rights reserved.
 
#pragma once
 
namespace ck {
namespace tensor_operation {
namespace device {
 
struct ArgumentSplitK
{
    index_t k_batch_{1};
};
 
} // namespace device
} // namespace tensor_operation
} // namespace ck