    def get_profile_cache_version(self, op_class: str) -> int:
        """Get the current profile cache version for the op_class.

        Parameters
        ----------
        op_class : str
            Op class name: only gemm is supported at the moment.

        Returns
        -------
        int
            cache version.

        Raises
        ------
        NotImplementedError
            If op class is not supported, raise error.
        """
        # TODO: support conv and normalization
        if op_class == "gemm":
            return self._profile_cache.gemm_cache_version
        elif op_class == "conv":
            return self._profile_cache.conv_cache_version
        elif op_class == "conv3d":
            return self._profile_cache.conv3d_cache_version
        raise NotImplementedError
Get the current profile cache version for the op_class.

Parameters
----------
op_class : str
Op class name: only gemm is supported at the moment.

Returns
-------
int
cache version.

Raises
------
NotImplementedError
If op class is not supported, raise error.
python/aitemplate/backend/target.py