Enable more Inductor UT for XPU (#171773)

Summary:
X-link: pytorch/pytorch#171773
Approved by: https://github.com/zou3519

Reviewed By: atalman, jeanschmidt

Differential Revision: D90412033

fbshipit-source-id: 252c794818e7006bf51d9e991258554b4e8f966b

Author: generatedunixname499836121

userbenchmark/dynamo/dynamobench/_dynamo/utils.py

@@ -2335,13 +2335,17 @@ def preserve_rng_state() -> Generator[None, None, None]:
         skip_frame_if_in_functorch_mode(rng_state)
         if torch.cuda.is_available():
             cuda_rng_state = torch.clone(torch.cuda.get_rng_state())
+        if torch.xpu.is_available():
+            xpu_rng_state = torch.clone(torch.xpu.get_rng_state())
     try:
         yield
     finally:
         with torch.utils._python_dispatch._disable_current_modes():
             torch.random.set_rng_state(rng_state)
             if torch.cuda.is_available():
                 torch.cuda.set_rng_state(cuda_rng_state)  # type: ignore[possibly-undefined]
+            if torch.xpu.is_available():
+                torch.xpu.set_rng_state(xpu_rng_state)  # type: ignore[possibly-undefined]
 
 
 def is_jit_model(