Add __device__ keyword.

Add device keyword.
991e44ee · Adam Osewski · fd3972ae · 991e44ee
Commit 991e44ee authored Oct 06, 2023 by Adam Osewski
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

include/ck/tensor_operation/gpu/grid/block_to_ctile_map.hpp include/ck/tensor_operation/gpu/grid/block_to_ctile_map.hpp +2 -2

No files found.
--- a/include/ck/tensor_operation/gpu/grid/block_to_ctile_map.hpp
+++ b/include/ck/tensor_operation/gpu/grid/block_to_ctile_map.hpp
@@ -1176,7 +1176,7 @@ struct BlockToCTileMap_LinearKSplit
    {
    }

-    __host__ constexpr index_t CalculateGridSize(index_t M, index_t N)
+    __host__ __device__ constexpr index_t CalculateGridSize(index_t M, index_t N)
    {
        const auto M0 = math::integer_divide_ceil(M, MPerBlock);
        const auto N0 = math::integer_divide_ceil(N, NPerBlock);
@@ -1185,7 +1185,7 @@ struct BlockToCTileMap_LinearKSplit
    }

    template <typename CGridDesc_M_N>
-    __host__ constexpr index_t CalculateGridSize(const CGridDesc_M_N& c_grid_desc_m_n)
+    __host__ __device__ constexpr index_t CalculateGridSize(const CGridDesc_M_N& c_grid_desc_m_n)
    {
        return CalculateGridSize(c_grid_desc_m_n.GetLength(I0), c_grid_desc_m_n.GetLength(I1));
    }