Files
composable_kernel/src/include/gridwise_convolution_wrapper.hip.hpp
2019-04-02 19:37:02 -05:00

11 lines
440 B
C++

#pragma once
template <class GridwiseConvolution, class T>
__global__ void run_gridwise_convolution(GridwiseConvolution,
const T* const __restrict__ p_in_global,
const T* const __restrict__ p_wei_global,
T* const __restrict__ p_out_global)
{
GridwiseConvolution{}.Run(p_in_global, p_wei_global, p_out_global);
}