"git@developer.sourcefind.cn:modelzoo/resnet50_tensorflow.git" did not exist on "af1a6c576671e9e4f4f58be6bfe2fcb513546322"
gridwise_operation_wrapper.hpp 336 Bytes
Newer Older
Chao Liu's avatar
Chao Liu committed
1
2
3
4
#ifndef CK_GRIDWISE_OPERATION_KERNEL_WRAPPER
#define CK_GRIDWISE_OPERATION_KERNEL_WRAPPER

template <typename GridwiseOp, typename... Xs>
Chao Liu's avatar
Chao Liu committed
5
6
7
8
9
__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)
Chao Liu's avatar
Chao Liu committed
10
11
12
13
14
{
    GridwiseOp{}.Run(xs...);
}

#endif