Composable kernel init integration v3 (#1097)

* Squashed 'src/composable_kernel/' content from commit f6edda611 git-subtree-dir: src/composable_kernel git-subtree-split: f6edda6119 * add solver ConvIgemmFwdV6r1DlopsNchwKcyxNkhw; rename static ck source files * Squashed 'src/composable_kernel/' changes from f6edda611..5781adf5c 5781adf5c Update develop (#5) (#6) 97e6d514f Merge pull request #4 from ROCmSoftwarePlatform/separate_online_compile 7b1ec41e5 refactor 49c33aaea refactor 54b3e73d1 rename git-subtree-dir: src/composable_kernel git-subtree-split: 5781adf5cf * fix * refactor * remove online compilation from CK * refactor * fix * add ctest * add c-style pointer cast * vector/scalar pointer cast use c-style pointer cast instead of reinterpret_cast * fix clang warning suppression * tidy * suppress cppcheck * fix enum issue * revert chagnes to hip build * fix kernel filename * update CK build script * rename * rename * make innner product compatiable on gfx900 * Update src/include/miopen/solver/ck_utility_common.hpp Co-authored-by: JD <Jehandad.Khan@amd.com> * compiler parameter use stream * use int instead of index_t in kernel wrapper * DynamicBuffer, StaticBuffer, amd_buffer_load support customized value for invalid element * refactor * refactor * change cmakelist * change ck common utility * fix Co-authored-by: JD <Jehandad.Khan@amd.com>
2026-05-12 09:16:52 +00:00 · 2021-08-19 10:55:03 -05:00
commit 6fe3627a9e
126 changed files with 30654 additions and 0 deletions
--- a/host/solver/include/convolution_problem_descriptor.hpp
+++ b/host/solver/include/convolution_problem_descriptor.hpp
@@ -0,0 +1,81 @@
+#ifndef CONVOLUTION_PROBLEM_DESCRIPTOR
+#define CONVOLUTION_PROBLEM_DESCRIPTOR
+
+namespace ck {
+namespace driver {
+
+struct ConvolutionProblemDescriptor
+{
+    ConvolutionProblemDescriptor() = default;
+
+    ConvolutionProblemDescriptor(int N_,
+                                 int K_,
+                                 int C_,
+                                 int Y_,
+                                 int X_,
+                                 int Hi_,
+                                 int Wi_,
+                                 int Ho_,
+                                 int Wo_,
+                                 int ConvStrideH_,
+                                 int ConvStrideW_,
+                                 int ConvDilationH_,
+                                 int ConvDilationW_,
+                                 int InLeftPadH_,
+                                 int InLeftPadW_,
+                                 int InRightPadH_,
+                                 int InRightPadW_,
+                                 ck::DataTypeEnum_t InDataTypeEnum_,
+                                 ck::DataTypeEnum_t WeiDataTypeEnum_,
+                                 ck::DataTypeEnum_t OutDataTypeEnum_)
+        : N{N_},
+          K{K_},
+          C{C_},
+          Y{Y_},
+          X{X_},
+          Hi{Hi_},
+          Wi{Wi_},
+          Ho{Ho_},
+          Wo{Wo_},
+          ConvStrideH{ConvStrideH_},
+          ConvStrideW{ConvStrideW_},
+          ConvDilationH{ConvDilationH_},
+          ConvDilationW{ConvDilationW_},
+          InLeftPadH{InLeftPadH_},
+          InLeftPadW{InLeftPadW_},
+          InRightPadH{InRightPadH_},
+          InRightPadW{InRightPadW_},
+          InDataTypeEnum{InDataTypeEnum_},
+          WeiDataTypeEnum{WeiDataTypeEnum_},
+          OutDataTypeEnum{OutDataTypeEnum_}
+    {
+    }
+
+    int N;
+    int K;
+    int C;
+    int Y;
+    int X;
+    int Hi;
+    int Wi;
+    int Ho;
+    int Wo;
+    int ConvStrideH;
+    int ConvStrideW;
+    int ConvDilationH;
+    int ConvDilationW;
+    int InLeftPadH;
+    int InLeftPadW;
+    int InRightPadH;
+    int InRightPadW;
+
+    ck::DataTypeEnum_t InDataTypeEnum;
+    ck::DataTypeEnum_t WeiDataTypeEnum;
+    ck::DataTypeEnum_t OutDataTypeEnum;
+
+    std::size_t CalculateFlop() const { return 2L * N * K * C * Y * X * Ho * Wo; }
+};
+
+} // namespace driver
+} // namespace ck
+#endif