From ecf40f558e4f338797232be7f3db1ea084262fe5 Mon Sep 17 00:00:00 2001 From: "Po-Yen, Chen" Date: Fri, 1 Jul 2022 01:08:36 +0800 Subject: [PATCH] Add missing include directives --- example/01_gemm/gemm_xdl_fp16.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/example/01_gemm/gemm_xdl_fp16.cpp b/example/01_gemm/gemm_xdl_fp16.cpp index 4b6c218a4b..e745ab550f 100644 --- a/example/01_gemm/gemm_xdl_fp16.cpp +++ b/example/01_gemm/gemm_xdl_fp16.cpp @@ -8,7 +8,9 @@ #include "ck/ck.hpp" #include "ck/tensor_operation/gpu/device/gemm_specialization.hpp" +#include "ck/tensor_operation/gpu/device/device_gemm_xdl.hpp" #include "ck/tensor_operation/gpu/device/device_gemm_xdl_cshuffle.hpp" +#include "ck/tensor_operation/gpu/device/device_gemm_xdl_producer_consumer_cshuffle.hpp" #include "ck/tensor_operation/gpu/element/element_wise_operation.hpp" #include "ck/library/utility/check_err.hpp"