From 78e7aae35942ad4cb61d0f1ac5a574c8debb1e72 Mon Sep 17 00:00:00 2001 From: OscarXu Date: Sat, 4 Jan 2025 14:06:05 +0800 Subject: [PATCH] clang check --- include/ck_tile/ops/fused_moe/kernel/fused_moegemm_kernel.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/include/ck_tile/ops/fused_moe/kernel/fused_moegemm_kernel.hpp b/include/ck_tile/ops/fused_moe/kernel/fused_moegemm_kernel.hpp index 0ff1d9e5d0..c8a6337399 100644 --- a/include/ck_tile/ops/fused_moe/kernel/fused_moegemm_kernel.hpp +++ b/include/ck_tile/ops/fused_moe/kernel/fused_moegemm_kernel.hpp @@ -383,7 +383,7 @@ struct FusedMoeGemmKernel auto o_window = [&]() { ODataType* o_ptr = reinterpret_cast(kargs.o_ptr); auto o_view_ = make_naive_tensor_view( + memory_operation_enum::atomic_add>( o_ptr, make_tuple(kargs.num_tokens, kargs.hidden_size), make_tuple(kargs.stride_token, 1),