Merge branch 'main' of https://github.com/NVIDIA/TransformerEngine

5b6ef054 · yuguo · 76060570 · a7eeb28b · 5b6ef054 · 5b6ef054
Commit 5b6ef054 authored Mar 17, 2025 by yuguo
20 changed files
--- a/tests/cpp/test_common.cu
+++ b/tests/cpp/test_common.cu
--- a/tests/cpp/test_common.h
+++ b/tests/cpp/test_common.h
--- a/tests/cpp/util/CMakeLists.txt
+++ b/tests/cpp/util/CMakeLists.txt
--- a/tests/cpp/util/test_nvrtc.cpp
+++ b/tests/cpp/util/test_nvrtc.cpp
--- a/tests/cpp/util/test_string.cpp
+++ b/tests/cpp/util/test_string.cpp
--- a/tests/jax/conftest.py
+++ b/tests/jax/conftest.py
--- a/tests/jax/distributed_test_base.py
+++ b/tests/jax/distributed_test_base.py
--- a/tests/jax/pytest.ini
+++ b/tests/jax/pytest.ini
--- a/tests/jax/test_custom_call_compute.py
+++ b/tests/jax/test_custom_call_compute.py
--- a/tests/jax/test_distributed_fused_attn.py
+++ b/tests/jax/test_distributed_fused_attn.py
--- a/tests/jax/test_distributed_layernorm.py
+++ b/tests/jax/test_distributed_layernorm.py
--- a/tests/jax/test_distributed_layernorm_mlp.py
+++ b/tests/jax/test_distributed_layernorm_mlp.py
--- a/tests/jax/test_distributed_softmax.py
+++ b/tests/jax/test_distributed_softmax.py
--- a/tests/jax/test_functions.py
+++ b/tests/jax/test_functions.py
--- a/tests/jax/test_fused_attn.py
+++ b/tests/jax/test_fused_attn.py
--- a/tests/jax/test_helper.py
+++ b/tests/jax/test_helper.py
--- a/tests/jax/test_layer.py
+++ b/tests/jax/test_layer.py
--- a/tests/jax/test_misc.py
+++ b/tests/jax/test_misc.py
--- a/tests/jax/test_praxis_layers.py
+++ b/tests/jax/test_praxis_layers.py
--- a/tests/jax/test_sanity_import.py
+++ b/tests/jax/test_sanity_import.py