include/ck/tensor_operation/gpu/device/impl/split_k_arg.hpp Source File

include/ck/tensor_operation/gpu/device/impl/split_k_arg.hpp Source File#

Composable Kernel: include/ck/tensor_operation/gpu/device/impl/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