vuiseng9 commited on
Commit
68b7258
·
1 Parent(s): cf5a2b3

Add unofficial TE wheel built from vuiseng9's fork: branch v2.5+nvfp4 (010e0427)

Browse files
readme.md CHANGED
@@ -3,6 +3,7 @@ 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
@@ -10,4 +11,15 @@ Targeting >= Hopper
10
  * cuda : 12.9
11
  * cudnn: 91002 (9.10.2)
12
  * TE stable v2.5 (CUDA_ARCHS="90;100;120")
 
 
 
 
 
 
 
 
 
 
 
13
  ```
 
3
 
4
  Targeting >= Hopper
5
 
6
+ ## Official Release, prebuilt wheel
7
  ```
8
  250823_a/transformer_engine-2.5.0+f05f12c9-cp312-cp312-linux_x86_64.whl
9
  * build environment: nvcr.io/nvidia/pytorch:25.06-py3
 
11
  * cuda : 12.9
12
  * cudnn: 91002 (9.10.2)
13
  * TE stable v2.5 (CUDA_ARCHS="90;100;120")
14
+ ```
15
+
16
+ ## Unofficial, prebuilt wheel for local experimentation
17
+ ```
18
+ unofficial/251015_a/transformer_engine-2.5.0+010e0427-cp312-cp312-linux_x86_64.whl
19
+ * build environment: 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: vuiseng9's fork: branch v2.5+nvfp4 (010e0427)
24
+ # CUDA_ARCHS="100"
25
  ```
unofficial/251015_a/transformer_engine-2.5.0+010e0427-cp312-cp312-linux_x86_64.whl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1aede90d2a7376ecfb0352b43e4f109f31d67aedcb3ac62262a010dbff3d77a5
3
+ size 150541317