Fix broken cutlass version comparison

Signed-off-by: Santi Villalba <[email protected]>

Author: sdvillal

op_builder/evoformer_attn.py

@@ -5,6 +5,7 @@
 
 from .builder import CUDAOpBuilder, installed_cuda_version
 import os
+from packaging.version import Version
 
 
 class EvoformerAttnBuilder(CUDAOpBuilder):
@@ -54,9 +55,7 @@ def is_compatible(self, verbose=False):
                 if verbose:
                     self.warning("Please pip install nvidia-cutlass if trying to pre-compile kernels")
                 return False
-            cutlass_major, cutlass_minor = cutlass.__version__.split('.')[:2]
-            cutlass_compatible = (int(cutlass_major) >= 3 and int(cutlass_minor) >= 1)
-            if not cutlass_compatible:
+            if Version(cutlass.__version__) < Version('3.1.0'):
                 if verbose:
                     self.warning("Please use CUTLASS version >= 3.1.0")
                 return False