mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-11 17:00:18 +00:00
6dfb92bbef
* 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>
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
|