ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-30 03:37:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixes problem descritpor

Browse Source
This commit is contained in:
Bartlomiej Kocot
2025-08-07 10:28:54 +00:00
parent 4d2ecedab2
commit 55a2a4cac7
5 changed files with 70 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
experimental/convolution_builder/convolution_example.cpp
View File

@@ -4,19 +4,21 @@
#include "convolution_builder.hpp"
// Example of solution description for Forward Conv with default settings
struct GroupedConvFwdXdlImplicitGemm : public GroupedConvBaseXdl {
// Example of problem description for Forward Conv with default settings
struct GroupedConvFwdXdlImplicitGemm : public GroupedConvBaseXdlV1 {
static constexpr ConvolutionDirection ConvolutionDirection_ = ConvolutionDirection::Forward;
};
// Example of solution description for Backward Weight Conv with default settings and Split K Two Stage
struct GroupedConvBwdWeightXdlImplicitGemm : public GroupedConvBaseXdl {
// Example of problem description for Backward Weight Conv with default settings and Split K Two Stage
struct GroupedConvBwdWeightXdlImplicitGemmTwoStage : public GroupedConvBaseXdlV1 {
static constexpr ConvolutionDirection ConvolutionDirection_ = ConvolutionDirection::BackwardWeight;
static constexpr SplitKSupport SplitKSupport_ = SplitKSupport::SupportedTwoStage;
};
int main () {
ConvolutionBuilder<GroupedConvFwdXdlImplicitGemm> builder;
std::cout << builder.GetKernelName() << std::endl;
ConvolutionBuilder<GroupedConvFwdXdlImplicitGemm> builder_fwd;
std::cout << builder_fwd.GetInstanceName() << std::endl;
ConvolutionBuilder<GroupedConvBwdWeightXdlImplicitGemmTwoStage> builder_bwd_weight_two_stage;
std::cout << builder_bwd_weight_two_stage.GetInstanceName() << std::endl;
return 0;
}