From 5e430f59a7a8b53b37732ecc335b50219c32583f Mon Sep 17 00:00:00 2001 From: YuhanXu Date: Mon, 12 Jan 2026 10:48:07 +0800 Subject: [PATCH] Fix CMakeLists.txt and conv_transpose_kernel_impl.h --- backends/metax_gpu/CMakeLists.txt | 7 ++++++- .../metax_gpu/kernels/impl/conv_transpose_kernel_impl.h | 3 +-- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/backends/metax_gpu/CMakeLists.txt b/backends/metax_gpu/CMakeLists.txt index ddadacdaf7c..f3aae6592ee 100755 --- a/backends/metax_gpu/CMakeLists.txt +++ b/backends/metax_gpu/CMakeLists.txt @@ -797,8 +797,13 @@ target_compile_definitions( ${TARGET_NAME} PUBLIC PADDLE_WITH_CUDA=1 PADDLE_WITH_CUSTOM_DEVICE=1 - mcblasContext=cublasContext + cublasContext=mcblasContext cublasLtContext=mcblasLtContext + cublasLtMatmulDescOpaque_t=mcblasLtMatmulDescOpaque_t + cublasLtMatrixLayoutOpaque_t=mcblasLtMatrixLayoutOpaque_t + cublasLtMatmulPreferenceOpaque_t=mcblasLtMatmulPreferenceOpaque_t + cublasLtMatmulAlgoOpaque_t=mcblasLtMatmulAlgoOpaque_t + cublasStatus_t=mcblasStatus_t GPUContext=CustomContext KPSContext=CustomContext STREAM_TYPE=cudaStream_t diff --git a/backends/metax_gpu/kernels/impl/conv_transpose_kernel_impl.h b/backends/metax_gpu/kernels/impl/conv_transpose_kernel_impl.h index aadc5d2b8a0..820072a7931 100644 --- a/backends/metax_gpu/kernels/impl/conv_transpose_kernel_impl.h +++ b/backends/metax_gpu/kernels/impl/conv_transpose_kernel_impl.h @@ -142,8 +142,7 @@ void ConvTransposeRawKernel(const Context& dev_ctx, (data_layout != DataLayout::kNHWC ? static_cast(out_dims[1]) / groups : static_cast(out_dims[out_dims.size() - 1]) / groups); - phi::funcs::Col2ImFunctor col2im; - phi::funcs::Col2VolFunctor col2vol; + phi::funcs::Col2ImFunctor col2im; funcs::ConcatFunctor concat_functor; // convolution transpose: gemm + col2im or col2vol (similar to conv-backward