Skip to content

Commit c7a9e95

Browse files
committed
[DOC] fix yapi issue
Signed-off-by: paulyu12 <[email protected]>
1 parent 1dee1eb commit c7a9e95

File tree

1 file changed

+8
-13
lines changed

1 file changed

+8
-13
lines changed

vllm_ascend/meta_registration.py

Lines changed: 8 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -80,23 +80,18 @@ def get_masked_input_and_mask_meta(input: torch.Tensor,
8080

8181
return masked_input, mask
8282

83-
def bgmv_expand_meta(x: torch.Tensor,
84-
weight: torch.Tensor,
85-
indices: torch.Tensor,
86-
y: torch.Tensor,
87-
slice_offset: int,
88-
slice_size: int):
83+
84+
def bgmv_expand_meta(x: torch.Tensor, weight: torch.Tensor,
85+
indices: torch.Tensor, y: torch.Tensor, slice_offset: int,
86+
slice_size: int):
8987

9088
y_out = torch.empty_like(y)
9189
return y_out
9290

93-
def sgmv_expand_meta(x: torch.Tensor,
94-
weight: torch.Tensor,
95-
lora_indices: torch.Tensor,
96-
seq_len: torch.Tensor,
97-
y: torch.Tensor,
98-
slice_offset: int,
99-
slice_size: int):
91+
92+
def sgmv_expand_meta(x: torch.Tensor, weight: torch.Tensor,
93+
lora_indices: torch.Tensor, seq_len: torch.Tensor,
94+
y: torch.Tensor, slice_offset: int, slice_size: int):
10095

10196
y_out = torch.empty_like(y)
10297
return y_out

0 commit comments

Comments
 (0)