mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-06-30 03:37:38 +00:00
Conv builder and solution descriptor
This commit is contained in:
22
experimental/convolution_builder/convolution_example.cpp
Normal file
22
experimental/convolution_builder/convolution_example.cpp
Normal 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;
|
||||
}
|
||||
Reference in New Issue
Block a user