gridwise_convolution_wrapper.hip.hpp

#pragma once

template <class GridwiseConvolution, class T>
__global__ void run_gridwise_convolution(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);
}