gridwise_convolution_wrapper.hip.hpp 440 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
#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);
}