Skip to content

Commit e0c3a51

Browse files
author
varun sundar rabindranath
committed
c_map zeros -> empty
Signed-off-by: varun sundar rabindranath <[email protected]>
1 parent 2e8f3ac commit e0c3a51

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/model_executor/layers/fused_moe/cutlass_moe.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -155,7 +155,7 @@ def cutlass_moe(
155155
a_map = torch.zeros((local_topk_ids.numel()),
156156
dtype=torch.int32,
157157
device=device)
158-
c_map = torch.zeros((local_topk_ids.numel()),
158+
c_map = torch.empty((local_topk_ids.numel()),
159159
dtype=torch.int32,
160160
device=device)
161161

0 commit comments

Comments
 (0)