From 44599229cd16decd135b7bbd08ad09e657f6f83f Mon Sep 17 00:00:00 2001
From: Atream <80757050+Atream@users.noreply.github.com>
Date: Wed, 19 Mar 2025 12:16:48 +0800
Subject: [PATCH] Update gate.py

---
 ktransformers/operators/gate.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ktransformers/operators/gate.py b/ktransformers/operators/gate.py
index 02848ac..271a144 100644
--- a/ktransformers/operators/gate.py
+++ b/ktransformers/operators/gate.py
@@ -125,7 +125,7 @@ class KMoEGate(BaseInjectedModule, KMoEGateBase):
 
 # adapted from https://github.com/vllm-project/vllm/blob/c77620d22d43daa7e0440e6267cbdd83f849ac64/vllm/model_executor/layers/fused_moe/fused_moe.py#L1071
 # This is used by the Deepseek-V2 and Deepseek-V3 model
-@torch.compile(dynamic=True)
+#@torch.compile(dynamic=True)
 def grouped_topk(hidden_states: torch.Tensor,
                  gating_output: torch.Tensor,
                  topk: int,