mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-14 10:09:41 +00:00
8592df2211
* conv3d compiles but has memory error
* conv3d works
* fix performance issue by using __builtin_amdgc_readfirstlane
* change MakeBlock2CTileMap to MakeDefaultBlock2CTileMap; change c_blockid_to* to cblockid_to*
* clang-format
* remove CK_EXPERIMENTAL_PASS_TENSOR_DECRIPTOR_BY_*; moved wrapper into DeviceConv3d
* format
* remove useless marc
* add comment
Co-authored-by: Chao Liu <chao.liu2@amd.com>
[ROCm/composable_kernel commit: 6dfb92bbef]
65 lines
768 B
C++
65 lines
768 B
C++
#ifndef TENSOR_LAYOUT_HPP
|
|
#define TENSOR_LAYOUT_HPP
|
|
|
|
namespace ck {
|
|
namespace tensor_layout {
|
|
|
|
struct BaseTensorLayout
|
|
{
|
|
};
|
|
|
|
namespace gemm {
|
|
|
|
struct RowMajor : public BaseTensorLayout
|
|
{
|
|
};
|
|
|
|
struct ColumnMajor : public BaseTensorLayout
|
|
{
|
|
};
|
|
} // namespace gemm
|
|
|
|
namespace convolution {
|
|
|
|
struct NHWC : public BaseTensorLayout
|
|
{
|
|
};
|
|
|
|
struct KYXC : public BaseTensorLayout
|
|
{
|
|
};
|
|
|
|
struct NHWK : public BaseTensorLayout
|
|
{
|
|
};
|
|
|
|
struct NCHW : public BaseTensorLayout
|
|
{
|
|
};
|
|
|
|
struct KCYX : public BaseTensorLayout
|
|
{
|
|
};
|
|
|
|
struct NKHW : public BaseTensorLayout
|
|
{
|
|
};
|
|
|
|
struct NDHWC : public BaseTensorLayout
|
|
{
|
|
};
|
|
|
|
struct KZYXC : public BaseTensorLayout
|
|
{
|
|
};
|
|
|
|
struct NDHWK : public BaseTensorLayout
|
|
{
|
|
};
|
|
|
|
} // namespace convolution
|
|
|
|
} // namespace tensor_layout
|
|
} // namespace ck
|
|
#endif
|