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(
204204 NUM_WARPS = 4
205205 NUM_CTAS = 1
206206 NUM_STAGES = 2
207- MAX_NREG = None
208207
209208 EVEN_K = K % BLOCK_K == 0 # type: ignore
210209
@@ -258,7 +257,6 @@ def _lora_expand(
258257 num_warps = NUM_WARPS ,
259258 num_ctas = NUM_CTAS ,
260259 num_stages = NUM_STAGES ,
261- maxnreg = MAX_NREG ,
262260 )
263261
264262 return
Original file line number Diff line number Diff line change @@ -168,7 +168,6 @@ def _lora_shrink(
168168 NUM_WARPS = 4
169169 NUM_CTAS = 1
170170 NUM_STAGES = 2
171- MAX_NREG = None
172171
173172 EVEN_K = K % (BLOCK_K * SPLIT_K ) == 0 # type: ignore
174173
@@ -213,7 +212,6 @@ def _lora_shrink(
213212 num_warps = NUM_WARPS ,
214213 num_ctas = NUM_CTAS ,
215214 num_stages = NUM_STAGES ,
216- maxnreg = MAX_NREG ,
217215 )
218216
219217 return
You can’t perform that action at this time.
0 commit comments