Commits · 97a09c29cffa36efd48cd57f765b0cdc02674eaf · OpenDAS / TransformerEngine

22 Dec, 2025 1 commit

Fix ptxas compilation on sm103 for triton kernels (#2539) · 97a09c29

Teddy Do authored Dec 22, 2025



* add triton ptxas path  for gb300 to find where it is to avoid compilation errors
Signed-off-by: tdophung <tdophung@nvidia.com>

* add these flags in advance to preven future breaks when ops are extended to multi gpus
Signed-off-by: tdophung <tdophung@nvidia.com>

* add this also to L1
Signed-off-by: tdophung <tdophung@nvidia.com>

---------
Signed-off-by: tdophung <tdophung@nvidia.com>

97a09c29

05 Dec, 2025 1 commit
- [JAX] Enable TE/JAX test timings in CI (#2475) · 50be0299
  jberchtold-nvidia authored Dec 04, 2025
```
enable jax test timings in qa scripts
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>
```
  50be0299
30 Oct, 2025 1 commit
- [JAX] Ensure JAX reference impl uses an accurate backend in our tests (#2322) · 1269b2e2
  jberchtold-nvidia authored Oct 30, 2025
```
Ensure JAX reference impl uses an accurate backend
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>
```
  1269b2e2
13 Aug, 2025 1 commit

[JAX] Add L2_jax_distributed_unittest (#2060) · ec65ba3c

jberchtold-nvidia authored Aug 12, 2025



* Add L2_jax_distributed_unittest
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>

* Add L1 entry for NORM_INPUT_SHAPES that was missing
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>

---------
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>

ec65ba3c