def _invert_exec_key(self, key):
        """Invert execution key to get input arguments as integers.

        Parameters
        ----------
        key : str
            Execution key

        Returns
        ----------
            Dict[str, int]
        """
        vals = []
        key_strs = []
        for item in re.split(" == | && ", key):
            if item.isnumeric():
                vals.append(int(item))
            else:
                key_strs.append(item.strip())
        assert len(vals) == len(
            key_strs
        ), f"expected len(vals) == len(key_strs), but got {len(vals)}, {len(key_strs)}"
        return dict(zip(key_strs, vals))
Invert execution key to get input arguments as integers.

        Parameters
        ----------
        key : str
            Execution key

        Returns
        ----------
            Dict[str, int]
        
python/aitemplate/compiler/ops/groupnorm/groupnorm.py