[PROTON][XPU] Enable test_cmd.py in CI (#5330)

Signed-off-by: Anatoly Myachev <[email protected]>

Author: anmyachev

.github/workflows/build-test-reusable.yml

@@ -302,7 +302,7 @@ jobs:
           export LD_LIBRARY_PATH=$PTI_LIBS_DIR:$LD_LIBRARY_PATH
           export TRITON_XPUPTI_LIB_PATH=$PTI_LIBS_DIR
           cd third_party/proton/test
-          pytest test_api.py test_lib.py test_profile.py test_viewer.py -s -v
+          pytest test_api.py test_cmd.py test_lib.py test_profile.py test_viewer.py -s -v
           cd ..
 
       - name: Run minicore tests

third_party/proton/test/helper.py

@@ -17,8 +17,8 @@ def test_main():
 
 
 def matmul():
-    a = torch.randn((32, 32), device="cuda", dtype=torch.float16)
-    b = torch.randn((32, 32), device="cuda", dtype=torch.float16)
+    a = torch.randn((32, 32), device="xpu", dtype=torch.float16)
+    b = torch.randn((32, 32), device="xpu", dtype=torch.float16)
     M, K = a.shape
     K, N = b.shape
     c = torch.empty((M, N), device=a.device, dtype=a.dtype)