vuiseng9 commited on
Commit
cf5a2b3
·
1 Parent(s): 131e145

Wipe all old build; Add TE v2.5

Browse files
.gitattributes CHANGED
@@ -33,6 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
- 250815-a/transformer_engine-2.5.0+af9467c6-cp312-cp312-linux_x86_64.whl filter=lfs diff=lfs merge=lfs -text
37
- 250821-a/transformer_engine-2.8.0.dev0+734bcedd-cp312-cp312-linux_x86_64.whl filter=lfs diff=lfs merge=lfs -text
38
  *.whl filter=lfs diff=lfs merge=lfs -text
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
36
  *.whl filter=lfs diff=lfs merge=lfs -text
250815-a/transformer_engine-2.5.0+af9467c6-cp312-cp312-linux_x86_64.whl DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:41d6224a0ec73ec989c2a48d8a8431bcb0932c1b93def5f5fc808fa13868f5e8
3
- size 83678194
 
 
 
 
250821-a/transformer_engine-2.8.0.dev0+734bcedd-cp312-cp312-linux_x86_64.whl DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:1ea260968287955f189b4b323358662823b102bd2196437fec68a336314f734c
3
- size 93959270
 
 
 
 
250626-a/transformer_engine-2.4.0+3cd6870c-cp312-cp312-linux_x86_64.whl → 250823_a/transformer_engine-2.5.0+f05f12c9-cp312-cp312-linux_x86_64.whl RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0acca3c69850d4a554f49476c8ac077ebdc341aa36c4cc9bef39a95f55042b77
3
- size 82550904
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0d2d2d2ae8cab65afdc757877e076abfed31fb977b4794a8031c8a2bece50cdf
3
+ size 132986343
readme.md CHANGED
@@ -1,24 +1,13 @@
1
 
 
2
 
3
- ```
4
- 250815-a/
5
- * nvcr.io/nvidia/pytorch:25.06-py3
6
- * torch: 2.8.0a0+5228986c39.nv25.06
7
- * cuda : 12.9
8
- * cudnn: 91002 (9.10.2)
9
- * TE stable v2.4
10
-
11
- 250815-a/transformer_engine-2.5.0+af9467c6-cp312-cp312-linux_x86_64.whl
12
- * nvcr.io/nvidia/pytorch:25.06-py3
13
- * torch: 2.8.0a0+5228986c39.nv25.06
14
- * cuda : 12.9
15
- * cudnn: 91002 (9.10.2)
16
- * TE stable v2.5
17
 
18
- 250821-a/transformer_engine-2.8.0.dev0+734bcedd-cp312-cp312-linux_x86_64.whl
19
- * nvcr.io/nvidia/pytorch:25.06-py3
20
- * torch: 2.8.0a0+5228986c39.nv25.06
21
- * cuda : 12.9
22
- * cudnn: 91002 (9.10.2)
23
- * TE commit 734bcedd (2.8.0.dev0)
 
24
  ```
 
1
 
2
+ Transformer Engine prebuilt Wheel
3
 
4
+ Targeting >= Hopper
 
 
 
 
 
 
 
 
 
 
 
 
 
5
 
6
+ ```
7
+ 250823_a/transformer_engine-2.5.0+f05f12c9-cp312-cp312-linux_x86_64.whl
8
+ * build environment: nvcr.io/nvidia/pytorch:25.06-py3
9
+ * torch: 2.8.0a0+5228986c39.nv25.06
10
+ * cuda : 12.9
11
+ * cudnn: 91002 (9.10.2)
12
+ * TE stable v2.5 (CUDA_ARCHS="90;100;120")
13
  ```