Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
test.sh 325 Bytes
Newer Older
Paweł Gadziński's avatar
[PyTorch] Add test for TRT integration + fix for mxfp8 export (#2083)  
Paweł Gadziński committed
1
2
3
4
5 
# Copyright (c) 2022-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
#
# See LICENSE for license information.

: ${TE_PATH:=/opt/transformerengine}
shengfangd's avatar
[QA] Add pytest xml report for all tests in qa folder that use pytest (#2169)  
shengfangd committed
6
7 
: ${XML_LOG_DIR:=/logs}
mkdir -p "$XML_LOG_DIR"
Paweł Gadziński's avatar
[PyTorch] Add test for TRT integration + fix for mxfp8 export (#2083)  
Paweł Gadziński committed
8
shengfangd's avatar
[QA] Add pytest xml report for all tests in qa folder that use pytest (#2169)  
shengfangd committed
9 
python3 -m pytest --tb=auto --junitxml=$XML_LOG_DIR/test_onnx_export.xml $TE_PATH/tests/pytorch/test_onnx_export.py