From 1dfcb249f1accf8c048a9eae89298c75c34dce54 Mon Sep 17 00:00:00 2001
From: Abhilash Majumder <30946547+abhilash1910@users.noreply.github.com>
Date: Wed, 19 Apr 2023 23:55:21 +0530
Subject: [PATCH] WA for xpu Generator

---
 intel_extension_for_deepspeed/xpu_accelerator.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/intel_extension_for_deepspeed/xpu_accelerator.py b/intel_extension_for_deepspeed/xpu_accelerator.py
index 329b666..08112d2 100644
--- a/intel_extension_for_deepspeed/xpu_accelerator.py
+++ b/intel_extension_for_deepspeed/xpu_accelerator.py
@@ -81,6 +81,10 @@ def initial_seed(self, seed):
 
     def default_generator(self, device_index):
         return torch.xpu.default_generators[device_index]
+    
+    #WA for xpu Generator in torch api    
+    def xpu_generator(self, device_index):
+        return torch.xpu.Generator(device=f'xpu:{device_index}')
 
     # Streams/Events
     @property