From 4ac7b894c707bf49c8dc92bf17ba7d7904600b1b Mon Sep 17 00:00:00 2001 From: lalala-sh Date: Wed, 19 Nov 2025 03:39:10 +0000 Subject: [PATCH] fix the build error of rtp --- csrc/ck_deepgemm/include/deepgemm.h | 11 ----------- csrc/ck_deepgemm/include/deepgemm_common.cuh | 10 ++++++++++ 2 files changed, 10 insertions(+), 11 deletions(-) diff --git a/csrc/ck_deepgemm/include/deepgemm.h b/csrc/ck_deepgemm/include/deepgemm.h index ac201ca969..5d1ea5a6d4 100644 --- a/csrc/ck_deepgemm/include/deepgemm.h +++ b/csrc/ck_deepgemm/include/deepgemm.h @@ -1,21 +1,10 @@ #pragma once // SPDX-License-Identifier: MIT // Copyright (C) 2025, Advanced Micro Devices, Inc. All rights reserved. -#include "flatmm_basic.hpp" #include "py_itfs_common.h" #include #include -template , - class scaleN = ck_tile::FlatmmScalePointer<-1>> -using m_grouped_flatmm_args = ck_tile::MaskedGroupedFlatmmHostArgs; -using ck_stream_config = ck_tile::stream_config; -using row_major = ck_tile::tensor_layout::gemm::RowMajor; -using col_major = ck_tile::tensor_layout::gemm::ColumnMajor; -using bf16 = ck_tile::bf16_t; -using fp16 = ck_tile::half_t; -using fp8 = ck_tile::fp8_t; - torch::Tensor deepgemm(torch::Tensor& XQ, torch::Tensor& WQ, torch::Tensor& Y, diff --git a/csrc/ck_deepgemm/include/deepgemm_common.cuh b/csrc/ck_deepgemm/include/deepgemm_common.cuh index f0beb029fa..ecc946b456 100644 --- a/csrc/ck_deepgemm/include/deepgemm_common.cuh +++ b/csrc/ck_deepgemm/include/deepgemm_common.cuh @@ -19,6 +19,16 @@ #include "ck_tile/ops/flatmm.hpp" #include "ck_tile/ops/gemm.hpp" #include "deepgemm.h" +#include "flatmm_basic.hpp" +template , + class scaleN = ck_tile::FlatmmScalePointer<-1>> +using m_grouped_flatmm_args = ck_tile::MaskedGroupedFlatmmHostArgs; +using ck_stream_config = ck_tile::stream_config; +using row_major = ck_tile::tensor_layout::gemm::RowMajor; +using col_major = ck_tile::tensor_layout::gemm::ColumnMajor; +using bf16 = ck_tile::bf16_t; +using fp16 = ck_tile::half_t; +using fp8 = ck_tile::fp8_t; template