Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
gridwise_operation_wrapper.hpp 336 Bytes
Newer Older
Chao Liu's avatar
Composable kernel init integration v3 (#1097)  
Chao Liu committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14 
#ifndef CK_GRIDWISE_OPERATION_KERNEL_WRAPPER
#define CK_GRIDWISE_OPERATION_KERNEL_WRAPPER

template <typename GridwiseOp, typename... Xs>
__global__ void
#if CK_USE_LAUNCH_BOUNDS
    __launch_bounds__(CK_MAX_THREAD_PER_BLOCK, CK_MIN_BLOCK_PER_CU)
#endif
        run_gridwise_operation(Xs... xs)
{
    GridwiseOp{}.Run(xs...);
}

#endif