From fe1ed764f716043546f854042ecb5fbaf1351c56 Mon Sep 17 00:00:00 2001 From: Thorsten Kurth Date: Tue, 6 Jan 2026 01:00:42 -0800 Subject: [PATCH] fixing stream behaviour for rb buffer Signed-off-by: Thorsten Kurth --- src/csrc/rl/off_policy/interface.cpp | 1 - 1 file changed, 1 deletion(-) diff --git a/src/csrc/rl/off_policy/interface.cpp b/src/csrc/rl/off_policy/interface.cpp index d7b5e6f..967dee5 100644 --- a/src/csrc/rl/off_policy/interface.cpp +++ b/src/csrc/rl/off_policy/interface.cpp @@ -183,7 +183,6 @@ torchfort_result_t torchfort_rl_off_policy_train_step(const char* name, float* p c10::cuda::OptionalCUDAStreamGuard stream_guard; c10::cuda::OptionalCUDAGuard cuda_guard; set_device_and_stream(stream_guard, cuda_guard, model_device, ext_stream); - set_device_and_stream(stream_guard, cuda_guard, rb_device, ext_stream); #endif try {