Conv builder and solution descriptor

This commit is contained in:
Bartlomiej Kocot
2025-08-06 13:45:02 -04:00
parent 594858dd6e
commit 4d2ecedab2
5 changed files with 164 additions and 1 deletions

View File

@@ -0,0 +1,22 @@
#include <iostream>
#include <hip/hip_runtime.h>
#include "convolution_builder.hpp"
// Example of solution description for Forward Conv with default settings
struct GroupedConvFwdXdlImplicitGemm : public GroupedConvBaseXdl {
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 {
static constexpr ConvolutionDirection ConvolutionDirection_ = ConvolutionDirection::BackwardWeight;
static constexpr SplitKSupport SplitKSupport_ = SplitKSupport::SupportedTwoStage;
};
int main () {
ConvolutionBuilder<GroupedConvFwdXdlImplicitGemm> builder;
std::cout << builder.GetKernelName() << std::endl;
return 0;
}