From d7c58ab96eb04b0bd279cb32430799b35af4be03 Mon Sep 17 00:00:00 2001
From: Tianlei Wu <tlwu@microsoft.com>
Date: Sat, 1 Feb 2025 01:36:14 +0000
Subject: [PATCH] init logger

---
 .../tools/kernel_explorer/kernels/conftest.py    | 16 ++++++++++++++++
 .../kernel_explorer/kernels/kernel_explorer.py   |  3 ---
 .../azure-pipelines/linux-rocm-ci-pipeline.yml   |  2 +-
 3 files changed, 17 insertions(+), 4 deletions(-)
 create mode 100644 onnxruntime/python/tools/kernel_explorer/kernels/conftest.py

diff --git a/onnxruntime/python/tools/kernel_explorer/kernels/conftest.py b/onnxruntime/python/tools/kernel_explorer/kernels/conftest.py
new file mode 100644
index 0000000000..bf7065cdff
--- /dev/null
+++ b/onnxruntime/python/tools/kernel_explorer/kernels/conftest.py
@@ -0,0 +1,16 @@
+# -------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License.
+# --------------------------------------------------------------------------
+import pytest
+
+@pytest.fixture(scope="session", autouse=True)
+def setup_once():
+
+    print("Setup before all tests.")
+    from kernel_explorer import set_ort_severity
+    set_ort_severity(3)
+
+    yield
+
+    print("Teardown after all tests.")
diff --git a/onnxruntime/python/tools/kernel_explorer/kernels/kernel_explorer.py b/onnxruntime/python/tools/kernel_explorer/kernels/kernel_explorer.py
index 7568bc5873..8e645df373 100644
--- a/onnxruntime/python/tools/kernel_explorer/kernels/kernel_explorer.py
+++ b/onnxruntime/python/tools/kernel_explorer/kernels/kernel_explorer.py
@@ -365,6 +365,3 @@ def set_dispatchable_pattern(p: str = "*"):
 
 def set_return_tuning_results(b: bool = True):
     _ke_context.return_tuning_results = b
-
-# set default logger severity once so that we can create the defautl logger.
-set_ort_severity(3)
diff --git a/tools/ci_build/github/azure-pipelines/linux-rocm-ci-pipeline.yml b/tools/ci_build/github/azure-pipelines/linux-rocm-ci-pipeline.yml
index 6ee9a8dbcd..8bf7754b50 100644
--- a/tools/ci_build/github/azure-pipelines/linux-rocm-ci-pipeline.yml
+++ b/tools/ci_build/github/azure-pipelines/linux-rocm-ci-pipeline.yml
@@ -189,7 +189,7 @@ jobs:
             set -ex; \
             env; \
             cd /onnxruntime_src/onnxruntime/python/tools/kernel_explorer/kernels; \
-            pytest -n 4 --reruns 1 --durations=100"
+            pytest -s -n 4 --reruns 1 --durations=100"
       workingDirectory: $(Build.SourcesDirectory)
     displayName: 'Run kernel explorer tests'
     condition: succeededOrFailed()