Add __device__ keyword.

Add device keyword.
5b9ab343 · Adam Osewski · 4a029960 · 5b9ab343
Commit 5b9ab343 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
@@ -1165,7 +1165,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);
@@ -1174,7 +1174,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));
    }