[None][fix] Fix cute dsl nvfp4 gemm autotune issue (#8761)

Signed-off-by: Mindy Li <[email protected]>
Co-authored-by: Kaiyu Xie <[email protected]>

Author: limin2021

tensorrt_llm/_torch/custom_ops/cute_dsl_custom_ops.py

@@ -49,6 +49,15 @@ def __init__(self, alpha: float, output_dtype: torch.dtype):
                     f"SM version {get_sm_version()} is not supported for CuteDSLNVFP4BlackwellLinear, it only supports SM 100"
                 )
 
+        # rewrite the hash function because the value of self.alpha doesn't affect the tactic.
+        def __hash__(self):
+            return hash((self.output_dtype, ))
+
+        def __eq__(self, other):
+            if not isinstance(other, CuteDSLNVFP4BlackwellLinear):
+                return False
+            return self.output_dtype == other.output_dtype
+
         def get_valid_tactics(
             self,
             inputs: List[torch.Tensor],