batched_contraction.hpp Source File

• include
• ck_tile
• ops

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 Tensor Contraction Operations.

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

utils.hpp