From 9d8f1e4c21ddb614ee35e2421eeb591ddc862a01 Mon Sep 17 00:00:00 2001 From: apoorva Date: Tue, 13 May 2025 10:02:45 +0000 Subject: [PATCH] Fixed the gemm specialization error --- example/01_gemm/gemm_wmma_bf16_v3.cpp | 5 +++-- example/01_gemm/gemm_wmma_f16_v3.cpp | 4 ++-- example/01_gemm/gemm_wmma_f8_bf16_v3.cpp | 2 +- 3 files changed, 6 insertions(+), 5 deletions(-) diff --git a/example/01_gemm/gemm_wmma_bf16_v3.cpp b/example/01_gemm/gemm_wmma_bf16_v3.cpp index 9f78e7632e..28b2123619 100644 --- a/example/01_gemm/gemm_wmma_bf16_v3.cpp +++ b/example/01_gemm/gemm_wmma_bf16_v3.cpp @@ -19,7 +19,7 @@ using AElementOp = PassThrough; using BElementOp = PassThrough; using CElementOp = PassThrough; -static constexpr auto GemmDefault = ck::tensor_operation::device::GemmSpecialization::GemmSpec; +static constexpr auto GemmDefault = ck::tensor_operation::device::GemmSpecialization::Default; // clang-format off using DeviceGemmInstance = ck::tensor_operation::device::DeviceGemm_Wmma_CShuffleV3< @@ -55,4 +55,5 @@ using ReferenceGemmInstanceGPU = ck::tensor_operation::device::ReferenceGemm