[Nightly] Add op_ut for windows UT test

.github/workflows/_windows_ut.yml

@@ -74,7 +74,7 @@ jobs:
   ut_test:
     runs-on: ${{ inputs.runner }}
     if: ${{ inputs.src_changed == 'true' || inputs.has_label == 'true' }}
-    timeout-minutes: 900
+    timeout-minutes: 2000
     steps:
       - name: Checkout torch-xpu-ops
         uses: actions/checkout@v4
@@ -90,6 +90,7 @@ jobs:
           echo "C:\ProgramData\miniforge3\Scripts" >> "$GITHUB_PATH"
           echo "C:\ProgramData\miniforge3\Library\bin" >> "$GITHUB_PATH"
           call "C:\ProgramData\miniforge3\Scripts\activate.bat"
+          rmdir /s /q "C:\ProgramData\miniforge3\envs\windows_ci"
           call conda clean -ay
           call conda remove --all -y -n windows_ci
           call conda create -n windows_ci python=${{ inputs.python }} cmake ninja -y
@@ -263,6 +264,21 @@ jobs:
 
           if not exist "%GITHUB_WORKSPACE%\ut_log" mkdir "%GITHUB_WORKSPACE%\ut_log"
           copy test_xpu.xml %GITHUB_WORKSPACE%\ut_log /Y
+      - name: Run OP UT
+        if: contains(inputs.ut, 'op_ut') || github.event_name == 'schedule'
+        shell: cmd
+        continue-on-error: true
+        run: |
+          call "C:\ProgramData\miniforge3\Scripts\activate.bat"
+          call "C:\Program Files (x86)\Microsoft Visual Studio\2022\BuildTools\VC\Auxiliary\Build\vcvars64.bat"
+          call conda activate windows_ci
+          set PYTORCH_TEST_WITH_SLOW=1
+          set PYTORCH_ENABLE_XPU_FALLBACK=1
+          set PYTEST_ADDOPTS=-v --timeout 600 --timeout_method=thread --max-worker-restart 1000000 -n 1
+          cd ../pytorch/third_party/torch-xpu-ops/test/xpu/
+          python run_test_with_windows_nighltly.py
+          if not exist "%GITHUB_WORKSPACE%\ut_log" mkdir "%GITHUB_WORKSPACE%\ut_log"
+          for /r . %%f in (op_ut_with_*.xml) do move "%%f" "%GITHUB_WORKSPACE%\ut_log\" >nul 2>&1 || echo "File move completed"
       - name: UT Test Results Summary
         shell: cmd
         run: |

.github/workflows/nightly_ondemand.yml

@@ -210,7 +210,7 @@ jobs:
     secrets: inherit
     uses: ./.github/workflows/_windows_ut.yml
     with:
-      ut: 'op_extended,test_xpu'
+      ut: 'op_extended,test_xpu,op_ut'
       python: ${{ needs.Conditions-Filter.outputs.python }}
       src_changed: false
       has_label: true

test/xpu/run_test_with_windows_nighltly.py

@@ -0,0 +1,133 @@
+import os
+import sys
+
+from skip_list_common import skip_dict
+from skip_list_win import skip_dict as skip_dict_win
+from skip_list_win_lnl import skip_dict as skip_dict_win_lnl
+
+res = 0
+IS_WINDOWS = sys.platform == "win32"
+
+
+def launch_test(test_case, skip_list=None, skip_files=None, exe_list=None):
+    os.environ["PYTORCH_TEST_WITH_SLOW"] = "1"
+    skip_options = ""
+    if skip_list is not None:
+        skip_options = ' -k "not ' + skip_list[0]
+        for skip_case in skip_list[1:]:
+            skip_option = " and not " + skip_case
+            skip_options += skip_option
+        skip_options += '"'
+
+        if skip_files is not None:
+            for skip_file in skip_files:
+                skip_options += f" --ignore={skip_file}"
+
+        test_command = (
+            f"pytest --junit-xml=./op_ut_with_windows_nightly.{test_case}.xml --max-worker-restart=1000 "
+            + test_case
+        )
+        test_command += skip_options
+    elif exe_list is not None:
+        exe_options = ' -k "' + exe_list[0]
+        for exe_case in exe_list[1:]:
+            exe_option = " or " + exe_case
+            exe_options += exe_option
+        exe_options += '"'
+
+        if skip_files is not None:
+            for skip_file in skip_files:
+                skip_options += f" --ignore={skip_file}"
+
+        test_command = (
+            f"pytest --junit-xml=./op_ut_with_windows_nightly.{test_case}.xml --max-worker-restart=1000 "
+            + test_case
+        )
+        test_command += exe_options
+    else:
+        if skip_files is not None:
+            for skip_file in skip_files:
+                skip_options += f" --ignore={skip_file}"
+
+        test_command = (
+            f"pytest --junit-xml=./op_ut_with_windows_nightly.{test_case}.xml --max-worker-restart=1000 "
+            + test_case
+        )
+        test_command += skip_options
+    return os.system(test_command)
+
+
+skip_files_list = [
+    "test_autocast_xpu.py",
+    "test_autograd_fallback_xpu.py",
+    "test_autograd_xpu.py",
+    # "test_binary_ufuncs_xpu.py",
+    "test_comparison_utils_xpu.py",
+    "test_complex_xpu.py",
+    "test_content_store_xpu.py",
+    "test_dataloader_xpu.py",
+    "test_decomp.py",
+    "test_decomp_xpu.py",
+    "test_distributions_xpu.py",
+    "test_dynamic_shapes_xpu.py",
+    "test_foreach_xpu.py",
+    # "test_indexing_xpu.py",
+    "test_linalg_xpu.py",
+    "test_maskedtensor_xpu.py",
+    # "test_masked_xpu.py",
+    "test_matmul_cuda_xpu.py",
+    "test_meta_xpu.py",
+    # "test_modules_xpu.py",
+    "test_namedtensor_xpu.py",
+    "test_native_functions_xpu.py",
+    "test_native_mha_xpu.py",
+    "test_nestedtensor_xpu.py",
+    "test_nn_xpu.py",
+    "test_ops_fwd_gradients_xpu.py",
+    # "test_ops_gradients_xpu.py",
+    # "test_ops_xpu.py",
+    "test_optim_xpu.py",
+    # "test_reductions_xpu.py",
+    # "test_scatter_gather_ops_xpu.py",
+    "test_segment_reductions_xpu.py",
+    "test_shape_ops_xpu.py",
+    "test_sort_and_select_xpu.py",
+    "test_sparse_csr_xpu.py",
+    "test_sparse_xpu.py",
+    "test_spectral_ops_xpu.py",
+    "test_tensor_creation_ops_xpu.py",
+    # "test_torch_xpu.py",
+    # "test_transformers_xpu.py",
+    "test_type_promotion_xpu.py",
+    # "test_unary_ufuncs_xpu.py",
+    # "test_view_ops_xpu.py",
+    "functorch/test_ops_xpu.py",
+]
+
+print("Current working directory:", os.getcwd())
+print("Files in directory:")
+for file in os.listdir("."):
+    if file.endswith(".py"):
+        print(f"  {file}")
+
+for key in skip_dict:
+    # Check if key is in skip list
+    if key in skip_files_list:
+        print(f"\n=== Skipping test file: {key} ===")
+        continue
+
+    skip_list = skip_dict[key]
+    skip_files = skip_files_list.copy()
+    if IS_WINDOWS and key in skip_dict_win:
+        skip_list += skip_dict_win[key]
+    if IS_WINDOWS and key in skip_dict_win_lnl:
+        skip_list += skip_dict_win_lnl[key]
+
+    print(f"\n=== Processing test case: {key} ===")
+    res += launch_test(key, skip_list, skip_files)
+
+if os.name == "nt":
+    sys.exit(res)
+else:
+    exit_code = os.WEXITSTATUS(res)
+    sys.exit(exit_code)