[Fix] Fix several bugs on DSA models (#22430)

2026-04-09 12:46:23 -07:00
parent 606aa11ea8
commit 60acdc31f2
2 changed files with 5 additions and 5 deletions
@@ -1491,8 +1491,10 @@ class ServerArgs:
            self.nsa_decode_backend = "tilelang"
        elif kv_cache_dtype == "fp8_e4m3":
            if major >= 10:
-                self.nsa_prefill_backend = "trtllm"
-                self.nsa_decode_backend = "trtllm"
+                if not user_set_prefill:
+                    self.nsa_prefill_backend = "trtllm"
+                if not user_set_decode:
+                    self.nsa_decode_backend = "trtllm"
            else:
                # flashmla_auto dispatches to flashmla_sparse/flashmla_kv based on hardware and heuristics
                if not user_set_prefill: