From bfc256ad748419104d6d5d3cfd633de1c2a5968a Mon Sep 17 00:00:00 2001
From: Ludovic Raess <ludovic.rass@gmail.com>
Date: Fri, 19 Jun 2026 14:28:31 +0200
Subject: [PATCH] Set stream before every execution

---
 src/fft/fft.jl | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/src/fft/fft.jl b/src/fft/fft.jl
index 1dfe213bd..17b3525f3 100644
--- a/src/fft/fft.jl
+++ b/src/fft/fft.jl
@@ -51,7 +51,6 @@ mutable struct cROCFFTPlan{T,K,inplace,N,R,B} <: ROCFFTPlan{T, K, inplace}
         rocfft_execution_info_create(info_ref)
         info = info_ref[]
 
-        # assign to the current stream
         stream = AMDGPU.stream()
         rocfft_execution_info_set_stream(info, stream)
         if length(workarea) > 0
@@ -99,11 +98,8 @@ end
 
 function update_stream!(plan::ROCFFTPlan)
     new_stream = AMDGPU.stream()
-    if plan.stream != new_stream
-        plan.stream = new_stream
-        info = plan.execution_info
-        rocfft_execution_info_set_stream(info, new_stream)
-    end
+    plan.stream = new_stream
+    rocfft_execution_info_set_stream(plan.execution_info, new_stream)
     return
 end