/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/batched_contraction.hpp Source File

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/batched_contraction.hpp Source File#

Composable Kernel: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/batched_contraction.hpp Source File
batched_contraction.hpp
Go to the documentation of this file.
1 // SPDX-License-Identifier: MIT
2 // Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.
3 
4 #pragma once
5 
6 #include "ck_tile/ops/batched_contraction/kernel/batched_contraction_kernel.hpp"
7 #include "ck_tile/ops/batched_contraction/pipeline/batched_contraction_problem.hpp"
8 #include "ck_tile/ops/batched_contraction/utils/tensor_descriptor_utils.hpp"
9 #include "ck_tile/ops/common/generic_2d_block_shape.hpp"
10 #include "ck_tile/ops/common/load_interleaved_pk_type.hpp"
11 #include "ck_tile/ops/common/streamk_common.hpp"
12 #include "ck_tile/ops/common/tensor_layout.hpp"
13 #include "ck_tile/ops/common/utils.hpp"
batched_contraction_kernel.hpp
batched_contraction_problem.hpp
generic_2d_block_shape.hpp
load_interleaved_pk_type.hpp
streamk_common.hpp
tensor_descriptor_utils.hpp
Utility functions for creating tensor descriptors in batched contraction operations.
tensor_layout.hpp