Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
test.sh 250 Bytes
Newer Older
Przemyslaw Tredak's avatar
Change the copyright to include 2024 (#583)  
Przemyslaw Tredak committed
1 
# Copyright (c) 2022-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
Alp Dener's avatar
[JAX] Regression tests for custom ops with jax.experimental.custom_partitioning (#471)  
Alp Dener committed
2
3
4
5
6
7 
#
# See LICENSE for license information.

set -xe

: ${TE_PATH:=/opt/transformerengine}
Alp Dener's avatar
[JAX] Fixing CI failure due to incorrect use of `static_argnums` in jax.jit (#785)  
Alp Dener committed
8 
pytest -c $TE_PATH/tests/jax/pytest.ini -v $TE_PATH/tests/jax/test_distributed_*