From 06994f2a132bd964f8ebe09279cb5ab44db4ed57 Mon Sep 17 00:00:00 2001
From: KakaruHayate <kakaru.hk@hotmail.com>
Date: Sun, 19 Oct 2025 16:49:13 +0800
Subject: [PATCH] may fix RuntimeError: MUSA error: an illegal memory access
 was encountered

---
 GPT_SoVITS/s2_train.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/GPT_SoVITS/s2_train.py b/GPT_SoVITS/s2_train.py
index 7d504edd..cb978181 100644
--- a/GPT_SoVITS/s2_train.py
+++ b/GPT_SoVITS/s2_train.py
@@ -17,6 +17,7 @@ import torch.multiprocessing as mp
 from torch.cuda.amp import GradScaler
 musa_ddp = False
 if musa_utils.is_available():
+    os.environ["MUSA_LAUNCH_BLOCKING"] = 1
     autocast = torch.musa.amp.autocast
     musa_ddp = musa_utils.should_ddp()
 elif torch.cuda.is_available():