Refactor the design of DeviceGemmMultipleDMultipleR_Xdl_CShuffle (#378)

[ROCm/composable_kernel commit: 88e43744d8]
2026-05-21 13:29:20 +00:00 · 2022-08-24 23:12:54 +08:00
parent 54af9c3aa0
commit 32ceea7b1c
7 changed files with 123 additions and 261 deletions
--- a/library/include/ck/library/tensor_operation_instance/gpu/gemm.hpp
+++ b/library/include/ck/library/tensor_operation_instance/gpu/gemm.hpp
@@ -4,6 +4,8 @@
 #pragma once

 #include <cstdlib>
+#include <memory>
+#include <vector>

 #include "ck/ck.hpp"
 #include "ck/tensor_operation/gpu/device/tensor_layout.hpp"