mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-12 01:10:17 +00:00
11 lines
440 B
C++
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);
|
|
}
|