Skip to content

Commit 46d1628

Browse files
committed
Formatting fix
1 parent c57b290 commit 46d1628

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/tvm/relax/frontend/nn/llm/position_embedding.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -571,7 +571,7 @@ def fused_rope_longrope_scaling( # pylint: disable=too-many-locals
571571
long_factors = T.Buffer((rotary_dim // 2,), "float32", data=ext_factors.data)
572572
short_factors = T.Buffer((rotary_dim // 2,), "float32", data=ext_factors.data, elem_offset=(rotary_dim // 2)) # type: ignore
573573

574-
if(seq_len > original_max_position_embeddings):
574+
if seq_len > original_max_position_embeddings:
575575
for iters in T.grid(seq_len, fused_heads, head_dim):
576576
with T.block("llama_fused_rope"):
577577
s, h, d = T.axis.remap("SSS", iters)

0 commit comments

Comments
 (0)