From 3d06952a2bfbefc7465735ec7a77158dcfd3e557 Mon Sep 17 00:00:00 2001
From: valarLip <103567126+valarLip@users.noreply.github.com>
Date: Thu, 13 Mar 2025 15:11:59 +0800
Subject: [PATCH] hotfix fmoe build issue (#1976)

[ROCm/composable_kernel commit: 52b1cd7780f412be64bb2b08aa10b91a6f2bf26a]
---
 .../ck_tile/15_fused_moe/instances/fused_moe_api.cpp   | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/example/ck_tile/15_fused_moe/instances/fused_moe_api.cpp b/example/ck_tile/15_fused_moe/instances/fused_moe_api.cpp
index 466420f066..b7eaf5c6e1 100644
--- a/example/ck_tile/15_fused_moe/instances/fused_moe_api.cpp
+++ b/example/ck_tile/15_fused_moe/instances/fused_moe_api.cpp
@@ -72,8 +72,14 @@ float fused_moe(fused_moe_traits t, fused_moe_args a, const ck_tile::stream_conf
 
     float r = ck_tile::launch_kernel(
         s,
-        [=, &r0](const ck_tile::stream_config&) { r0 = fused_moesorting(t0, a0, s_sub); },
-        [=, &r1](const ck_tile::stream_config&) { r1 = fused_moegemm(t1, a1, s_sub); });
+        [=, &r0](const ck_tile::stream_config&) {
+            r0 = fused_moesorting(t0, a0, s_sub);
+            return hipPeekAtLastError() == hipSuccess;
+        },
+        [=, &r1](const ck_tile::stream_config&) {
+            r1 = fused_moegemm(t1, a1, s_sub);
+            return hipPeekAtLastError() == hipSuccess;
+        });
 
     // keep unsupported case return negative
     if(r0 < 0 || r1 < 0)