Skip to content

Commit 6d9530c

Browse files
committed
remove unused code
Signed-off-by: Neta Zmora <[email protected]>
1 parent cfab9b7 commit 6d9530c

File tree

2 files changed

+0
-13
lines changed

2 files changed

+0
-13
lines changed

tensorrt_llm/_torch/custom_ops/torch_custom_ops.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,6 @@ def bmm_out(a: torch.Tensor, b: torch.Tensor, out: torch.Tensor) -> None:
2727
from enum import IntEnum
2828

2929

30-
# Copied from csrc/nv_internal/tensorrt_llm/kernels/cutlass_kernels/include/common.hExpand commentComment on line R76Code has comments. Press enter to view.
3130
class ActivationType(IntEnum):
3231
Gelu = 0
3332
Relu = 1

tensorrt_llm/_torch/modules/fused_moe/routing.py

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -155,18 +155,6 @@ class RoutingMethodType(IntEnum):
155155
Unspecified = 5.
156156

157157

158-
class ActivationType(IntEnum):
159-
Gelu = 0
160-
Relu = 1
161-
Silu = 2
162-
Swiglu = 3
163-
Geglu = 4
164-
SwigluBias = 5
165-
Relu2 = 6
166-
Identity = 7
167-
InvalidType = 8
168-
169-
170158
class BaseMoeRoutingMethod(nn.Module):
171159

172160
def apply(self, _router_logits) -> (torch.Tensor, torch.Tensor):

0 commit comments

Comments
 (0)