include/ck/utility/get_shift.hpp Source File

• include
• ck
• utility

get_shift.hpp

Go to the documentation of this file.

1 // SPDX-License-Identifier: MIT

2 // Copyright (c) 2018-2023, Advanced Micro Devices, Inc. All rights reserved.

3 

4 #pragma once

5 

6 namespace ck {

7 

8 template <index_t N>

9 static constexpr __device__ index_t get_shift()

10 {

11  return (get_shift<N / 2>() + 1);

12 };

13 

14 template <>

15 constexpr __device__ index_t get_shift<1>()

16 {

17  return (0);

18 }

19 

20 } // namespace ck

ck

Definition: ck.hpp:264

ck::get_shift< 1 >

constexpr __device__ index_t get_shift< 1 >()

Definition: get_shift.hpp:15

ck::index_t

int32_t index_t

Definition: ck.hpp:289