Ubuntu commited on
Commit
2c8e98a
·
1 Parent(s): a3d4355

feat: new layout

Browse files
build/{lib.linux-x86_64-3.10 → 2.5.1+cu124}/dropout_layer_norm.cpython-310-x86_64-linux-gnu.so RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:aba674c175147bfdff6acb354745749070519df35f66522d71b2743aedc3b5a9
3
- size 26705096
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5a2a2612ebb3261909ec7ca8b85213e19f5ea128aeee93eca5640222d299969c
3
+ size 734115376
setup.py CHANGED
@@ -116,60 +116,60 @@ ext_modules.append(
116
  "ln_api.cpp",
117
  "ln_fwd_256.cu",
118
  "ln_bwd_256.cu",
119
- # "ln_fwd_512.cu",
120
- # "ln_bwd_512.cu",
121
- # "ln_fwd_768.cu",
122
- # "ln_bwd_768.cu",
123
- # "ln_fwd_1024.cu",
124
- # "ln_bwd_1024.cu",
125
- # "ln_fwd_1280.cu",
126
- # "ln_bwd_1280.cu",
127
- # "ln_fwd_1536.cu",
128
- # "ln_bwd_1536.cu",
129
- # "ln_fwd_2048.cu",
130
- # "ln_bwd_2048.cu",
131
- # "ln_fwd_2560.cu",
132
- # "ln_bwd_2560.cu",
133
- # "ln_fwd_3072.cu",
134
- # "ln_bwd_3072.cu",
135
- # "ln_fwd_4096.cu",
136
- # "ln_bwd_4096.cu",
137
- # "ln_fwd_5120.cu",
138
- # "ln_bwd_5120.cu",
139
- # "ln_fwd_6144.cu",
140
- # "ln_bwd_6144.cu",
141
- # "ln_fwd_7168.cu",
142
- # "ln_bwd_7168.cu",
143
- # "ln_fwd_8192.cu",
144
- # "ln_bwd_8192.cu",
145
- # "ln_parallel_fwd_256.cu",
146
- # "ln_parallel_bwd_256.cu",
147
- # "ln_parallel_fwd_512.cu",
148
- # "ln_parallel_bwd_512.cu",
149
- # "ln_parallel_fwd_768.cu",
150
- # "ln_parallel_bwd_768.cu",
151
- # "ln_parallel_fwd_1024.cu",
152
- # "ln_parallel_bwd_1024.cu",
153
- # "ln_parallel_fwd_1280.cu",
154
- # "ln_parallel_bwd_1280.cu",
155
- # "ln_parallel_fwd_1536.cu",
156
- # "ln_parallel_bwd_1536.cu",
157
- # "ln_parallel_fwd_2048.cu",
158
- # "ln_parallel_bwd_2048.cu",
159
- # "ln_parallel_fwd_2560.cu",
160
- # "ln_parallel_bwd_2560.cu",
161
- # "ln_parallel_fwd_3072.cu",
162
- # "ln_parallel_bwd_3072.cu",
163
- # "ln_parallel_fwd_4096.cu",
164
- # "ln_parallel_bwd_4096.cu",
165
- # "ln_parallel_fwd_5120.cu",
166
- # "ln_parallel_bwd_5120.cu",
167
- # "ln_parallel_fwd_6144.cu",
168
- # "ln_parallel_bwd_6144.cu",
169
- # "ln_parallel_fwd_7168.cu",
170
- # "ln_parallel_bwd_7168.cu",
171
- # "ln_parallel_fwd_8192.cu",
172
- # "ln_parallel_bwd_8192.cu",
173
  ],
174
  extra_compile_args={
175
  "cxx": ["-O3"] + generator_flag,
 
116
  "ln_api.cpp",
117
  "ln_fwd_256.cu",
118
  "ln_bwd_256.cu",
119
+ "ln_fwd_512.cu",
120
+ "ln_bwd_512.cu",
121
+ "ln_fwd_768.cu",
122
+ "ln_bwd_768.cu",
123
+ "ln_fwd_1024.cu",
124
+ "ln_bwd_1024.cu",
125
+ "ln_fwd_1280.cu",
126
+ "ln_bwd_1280.cu",
127
+ "ln_fwd_1536.cu",
128
+ "ln_bwd_1536.cu",
129
+ "ln_fwd_2048.cu",
130
+ "ln_bwd_2048.cu",
131
+ "ln_fwd_2560.cu",
132
+ "ln_bwd_2560.cu",
133
+ "ln_fwd_3072.cu",
134
+ "ln_bwd_3072.cu",
135
+ "ln_fwd_4096.cu",
136
+ "ln_bwd_4096.cu",
137
+ "ln_fwd_5120.cu",
138
+ "ln_bwd_5120.cu",
139
+ "ln_fwd_6144.cu",
140
+ "ln_bwd_6144.cu",
141
+ "ln_fwd_7168.cu",
142
+ "ln_bwd_7168.cu",
143
+ "ln_fwd_8192.cu",
144
+ "ln_bwd_8192.cu",
145
+ "ln_parallel_fwd_256.cu",
146
+ "ln_parallel_bwd_256.cu",
147
+ "ln_parallel_fwd_512.cu",
148
+ "ln_parallel_bwd_512.cu",
149
+ "ln_parallel_fwd_768.cu",
150
+ "ln_parallel_bwd_768.cu",
151
+ "ln_parallel_fwd_1024.cu",
152
+ "ln_parallel_bwd_1024.cu",
153
+ "ln_parallel_fwd_1280.cu",
154
+ "ln_parallel_bwd_1280.cu",
155
+ "ln_parallel_fwd_1536.cu",
156
+ "ln_parallel_bwd_1536.cu",
157
+ "ln_parallel_fwd_2048.cu",
158
+ "ln_parallel_bwd_2048.cu",
159
+ "ln_parallel_fwd_2560.cu",
160
+ "ln_parallel_bwd_2560.cu",
161
+ "ln_parallel_fwd_3072.cu",
162
+ "ln_parallel_bwd_3072.cu",
163
+ "ln_parallel_fwd_4096.cu",
164
+ "ln_parallel_bwd_4096.cu",
165
+ "ln_parallel_fwd_5120.cu",
166
+ "ln_parallel_bwd_5120.cu",
167
+ "ln_parallel_fwd_6144.cu",
168
+ "ln_parallel_bwd_6144.cu",
169
+ "ln_parallel_fwd_7168.cu",
170
+ "ln_parallel_bwd_7168.cu",
171
+ "ln_parallel_fwd_8192.cu",
172
+ "ln_parallel_bwd_8192.cu",
173
  ],
174
  extra_compile_args={
175
  "cxx": ["-O3"] + generator_flag,