ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-05 20:55:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add static_assert() to check pipeline type requirements

Browse Source
This commit is contained in:
Po-Yen, Chen
2022-07-07 11:10:39 +08:00
parent 4853ff1dd8
commit f0985f0a13
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
include/ck/tensor_operation/gpu/grid/gridwise_gemm_pipeline_v1.hpp
View File

@@ -357,10 +357,13 @@ constexpr auto GridwiseGemmPipeline_v1_Selector()
{
if constexpr(LoopSched == LoopScheduler::Default)
{
static_assert(std::is_default_constructible_v<GridwiseGemmPipeline_v1<NumPrefetch>>);
return GridwiseGemmPipeline_v1<NumPrefetch>{};
}
else if constexpr(LoopSched == LoopScheduler::Interwave)
{
static_assert(
std::is_default_constructible_v<GridwiseGemmPipelineInterwave_v1<NumPrefetch>>);
return GridwiseGemmPipelineInterwave_v1<NumPrefetch>{};
}
}