dimension.hpp 368 Bytes
Newer Older
Chao Liu's avatar
Chao Liu committed
1
2
3
4
5
6
7
8
#ifndef CK_DIMENSION_HPP
#define CK_DIMENSION_HPP

#include "common_header.hpp"

namespace ck {

template <index_t Length, index_t Stride>
Chao Liu's avatar
Chao Liu committed
9
struct NativeDimension
Chao Liu's avatar
Chao Liu committed
10
{
Chao Liu's avatar
Chao Liu committed
11
12
    __host__ __device__ static constexpr auto GetLength() { return Number<Length>{}; }

Chao Liu's avatar
Chao Liu committed
13
14
15
16
17
    __host__ __device__ static constexpr auto GetStride() { return Number<Stride>{}; }
};

} // namespace ck
#endif