From d3843d0ac0d5b32e67cece74e8baa707ee7b5fef Mon Sep 17 00:00:00 2001 From: Yanxing-Shi Date: Wed, 7 May 2025 14:19:49 +0000 Subject: [PATCH] fix cmake --- tile_engine/ops/gemm/CMakeLists.txt | 1 - tile_engine/ops/gemm/configs/default_config.json | 12 ++++-------- tile_engine/ops/gemm/gemm_host_api.hpp | 5 +++-- 3 files changed, 7 insertions(+), 11 deletions(-) diff --git a/tile_engine/ops/gemm/CMakeLists.txt b/tile_engine/ops/gemm/CMakeLists.txt index eb9a4c5c46..f78bf44e9a 100644 --- a/tile_engine/ops/gemm/CMakeLists.txt +++ b/tile_engine/ops/gemm/CMakeLists.txt @@ -23,7 +23,6 @@ endif() file(STRINGS ${CMAKE_CURRENT_BINARY_DIR}/gemm_instance_blobs.txt GEMM_CODEGEN_BLOBS) -message("test--------------------------------------------------") add_custom_command( OUTPUT ${GEMM_CODEGEN_BLOBS} COMMAND ${Python3_EXECUTABLE} ${CMAKE_CURRENT_LIST_DIR}/gemm_instance_builder.py diff --git a/tile_engine/ops/gemm/configs/default_config.json b/tile_engine/ops/gemm/configs/default_config.json index d726ba6add..acfcc03cac 100644 --- a/tile_engine/ops/gemm/configs/default_config.json +++ b/tile_engine/ops/gemm/configs/default_config.json @@ -11,8 +11,7 @@ }, "tile_k": { "values": [ - 64, - 32 + 64 ] }, "warp_m": { @@ -42,7 +41,7 @@ }, "warp_tile_k": { "values": [ - 16 + 8 ] }, "kPadM": { @@ -62,20 +61,17 @@ }, "pipeline": { "values": [ - "compv3", "mem" ] }, "scheduler": { "values": [ - "intrawave", - "interwave" + "intrawave" ] }, "epilogue": { "values": [ - "default", - "cshuffle" + "default" ] } } \ No newline at end of file diff --git a/tile_engine/ops/gemm/gemm_host_api.hpp b/tile_engine/ops/gemm/gemm_host_api.hpp index 7f1c518eba..4c3ae12a16 100644 --- a/tile_engine/ops/gemm/gemm_host_api.hpp +++ b/tile_engine/ops/gemm/gemm_host_api.hpp @@ -3,13 +3,14 @@ #pragma once -#include - #include #include #include #include #include + +#include + #include "ck_tile/ops/gemm.hpp" enum class Metric