/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck/tensor_operation/gpu/device/impl/split_k_arg.hpp Source File

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck/tensor_operation/gpu/device/impl/split_k_arg.hpp Source File#

Composable Kernel: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck/tensor_operation/gpu/device/impl/split_k_arg.hpp Source File
split_k_arg.hpp
Go to the documentation of this file.
1 // SPDX-License-Identifier: MIT
2 // Copyright (c) 2025, Advanced Micro Devices, Inc. All rights reserved.
3 
4 #pragma once
5 
6 namespace ck {
7 namespace tensor_operation {
8 namespace device {
9 
10 struct ArgumentSplitK
11 {
12  index_t k_batch_{1};
13 };
14 
15 } // namespace device
16 } // namespace tensor_operation
17 } // namespace ck
ck
Definition: ck.hpp:268
ck::index_t
int32_t index_t
Definition: ck.hpp:299
ck::tensor_operation::device::ArgumentSplitK
Definition: split_k_arg.hpp:11
ck::tensor_operation::device::ArgumentSplitK::k_batch_
index_t k_batch_
Definition: split_k_arg.hpp:12