Skip to content

Commit 8d7c3a5

Browse files
committed
remove fake impl for inplace op
Signed-off-by: Dongxu Yang <[email protected]>
1 parent dfbc4c2 commit 8d7c3a5

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

tensorrt_llm/_torch/custom_ops/cpp_custom_ops.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -249,10 +249,6 @@ def _(experts_ids: torch.Tensor, recv_rank_count_cumsum: torch.Tensor,
249249
ep_size: int):
250250
pass
251251

252-
@torch.library.register_fake("trtllm::moe_initialize_workspace")
253-
def _(all_workspaces: torch.Tensor, ep_rank: int, ep_size: int):
254-
pass
255-
256252
@torch.library.custom_op("trtllm::group_rms_norm_base",
257253
mutates_args=("outputs", ))
258254
def group_rms_norm_base(

0 commit comments

Comments
 (0)