File tree Expand file tree Collapse file tree 2 files changed +0
-4
lines changed Expand file tree Collapse file tree 2 files changed +0
-4
lines changed Original file line number Diff line number Diff line change @@ -204,7 +204,6 @@ def _lora_expand(
204
204
NUM_WARPS = 4
205
205
NUM_CTAS = 1
206
206
NUM_STAGES = 2
207
- MAX_NREG = None
208
207
209
208
EVEN_K = K % BLOCK_K == 0 # type: ignore
210
209
@@ -258,7 +257,6 @@ def _lora_expand(
258
257
num_warps = NUM_WARPS ,
259
258
num_ctas = NUM_CTAS ,
260
259
num_stages = NUM_STAGES ,
261
- maxnreg = MAX_NREG ,
262
260
)
263
261
264
262
return
Original file line number Diff line number Diff line change @@ -168,7 +168,6 @@ def _lora_shrink(
168
168
NUM_WARPS = 4
169
169
NUM_CTAS = 1
170
170
NUM_STAGES = 2
171
- MAX_NREG = None
172
171
173
172
EVEN_K = K % (BLOCK_K * SPLIT_K ) == 0 # type: ignore
174
173
@@ -213,7 +212,6 @@ def _lora_shrink(
213
212
num_warps = NUM_WARPS ,
214
213
num_ctas = NUM_CTAS ,
215
214
num_stages = NUM_STAGES ,
216
- maxnreg = MAX_NREG ,
217
215
)
218
216
219
217
return
You can’t perform that action at this time.
0 commit comments