From 9e244372e52ba75b5854349b18a0a17dea1978b9 Mon Sep 17 00:00:00 2001
From: Ivecia <sen@mail.ustc.edu.cn>
Date: Tue, 4 Nov 2025 13:24:48 +0800
Subject: [PATCH] [Fix] Fix potential deadlock caused by the last `EXIT`
 instruction.

---
 gpu-simulator/trace-driven/trace_driven.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gpu-simulator/trace-driven/trace_driven.cc b/gpu-simulator/trace-driven/trace_driven.cc
index 2c21acc14..f27901c3e 100644
--- a/gpu-simulator/trace-driven/trace_driven.cc
+++ b/gpu-simulator/trace-driven/trace_driven.cc
@@ -585,7 +585,7 @@ const warp_inst_t *trace_shader_core_ctx::get_next_inst(unsigned warp_id,
   trace_shd_warp_t *m_trace_warp =
       static_cast<trace_shd_warp_t *>(m_warp[warp_id]);
   const trace_warp_inst_t *ret = m_trace_warp->get_next_trace_inst();
-  if (m_trace_warp->trace_done()) {
+  if (ret == NULL && m_trace_warp->trace_done()) {
     if (!m_warp[warp_id]->inst_in_pipeline() &&
         m_warp[warp_id]->stores_done() &&
         !m_scoreboard->pendingWrites(warp_id)) {