puting gridwise convolution into its own class

This commit is contained in:
Chao Liu
2019-04-02 19:37:02 -05:00
parent bdbc0eaad1
commit 0b41ca2d9e
3 changed files with 24 additions and 16 deletions

View File

@@ -0,0 +1,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);
}