/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/batched_transpose/kernel/batched_transpose_kernel.hpp Source File#
batched_transpose_kernel.hpp
Go to the documentation of this file.
Definition: cluster_descriptor.hpp:13
constexpr CK_TILE_HOST_DEVICE auto integer_divide_ceil(X x, Y y)
Definition: math.hpp:149
constexpr CK_TILE_HOST_DEVICE auto pad_tensor_view(const TensorView &tensor_view, const TileLengths &tile_lengths, DoPads)
Definition: tensor_view.hpp:530
remove_cv_t< std::remove_reference_t< T > > remove_cvref_t
Definition: type_traits.hpp:21
constexpr CK_TILE_DEVICE auto make_tile_window(null_tensor_view, const WindowLengths &window_lengths, const multi_index< WindowLengths::size()> &, Ts &&...)
Definition: null_tile_window.hpp:75
constexpr CK_TILE_HOST_DEVICE auto make_tuple(Xs &&... xs)
Definition: tuple.hpp:360
Definition: batched_transpose_kernel.hpp:16
void * p_output
Definition: batched_transpose_kernel.hpp:18
index_t dim_block_w
Definition: batched_transpose_kernel.hpp:24
index_t dim_stride
Definition: batched_transpose_kernel.hpp:22
const void * p_input
Definition: batched_transpose_kernel.hpp:17
index_t dim_block_h
Definition: batched_transpose_kernel.hpp:23
Definition: batched_transpose_kernel.hpp:40
index_t width
Definition: batched_transpose_kernel.hpp:45
index_t height
Definition: batched_transpose_kernel.hpp:44
index_t dim_stride
Definition: batched_transpose_kernel.hpp:46
index_t batch
Definition: batched_transpose_kernel.hpp:43
const void * p_input
Definition: batched_transpose_kernel.hpp:41
void * p_output
Definition: batched_transpose_kernel.hpp:42
Definition: batched_transpose_kernel.hpp:29
static constexpr CK_TILE_HOST auto GridSize(const Hargs &host_args)
Definition: batched_transpose_kernel.hpp:52
remove_cvref_t< typename Pipeline::Problem > Problem
Definition: batched_transpose_kernel.hpp:33
CK_TILE_DEVICE void operator()(Kargs kargs) const
Definition: batched_transpose_kernel.hpp:76
static constexpr CK_TILE_HOST auto BlockSize()
Definition: batched_transpose_kernel.hpp:74
static constexpr CK_TILE_HOST auto MakeKargs(const Hargs &h)
Definition: batched_transpose_kernel.hpp:62
typename Problem::DataType Type
Definition: batched_transpose_kernel.hpp:35
static CK_TILE_DEVICE index_t counter
Definition: batched_transpose_kernel.hpp:31
remove_cvref_t< Pipeline_ > Pipeline
Definition: batched_transpose_kernel.hpp:32
static constexpr index_t kBlockSize
Definition: batched_transpose_kernel.hpp:37
Definition: integral_constant.hpp:13
Definition: coordinate_transform.hpp:1392
Definition: sequence.hpp:49