Skip to content

Commit a10d262

Browse files
committed
Fix issue
1 parent af55e1c commit a10d262

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

torch/_inductor/kernel/flex/templates/flex_backwards.py.jinja

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -231,8 +231,6 @@
231231
start_n1 = pid * BLOCK_N1
232232
offs_n1 = start_n1 + tl.arange(0, BLOCK_N1)
233233

234-
desc_k = None
235-
desc_v = None
236234
{%- if USE_TMA %}
237235
desc_k = tl.make_tensor_descriptor(
238236
base=K,
@@ -246,8 +244,6 @@
246244
strides=(stride_vn, stride_vd),
247245
block_shape=[BLOCK_N1, V_HEAD_DIM_ROUNDED],
248246
)
249-
{%- endif %}
250-
{%- if USE_TMA %}
251247
k = tl.load_tensor_descriptor(
252248
desc_k,
253249
[start_n1.to(tl.int32), 0],

0 commit comments

Comments
 (0)