saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-23 19:32:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

init logger

Browse source
This commit is contained in:
Tianlei Wu 2025-02-01 01:36:14 +00:00
parent 7a73fa986e
commit d7c58ab96e
3 changed files with 17 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
onnxruntime/python/tools/kernel_explorer/kernels/conftest.py Normal file
View file

@ -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.")

3
onnxruntime/python/tools/kernel_explorer/kernels/kernel_explorer.py
View file

@ -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)

2
tools/ci_build/github/azure-pipelines/linux-rocm-ci-pipeline.yml
View file

@ -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()