    def __init__(
        self,
        template_path=CUTLASS_PATH,
        ait_static_files_path=AIT_STATIC_FILES_PATH,
        arch="80",
        cuda_version=None,
        **kwargs,
    ):
        """CUDA target init.

        Parameters
        ----------
        template_path : str, optional
            by default "${repo_root}/3rdparty/cutlass"
        ait_static_files_path : str
            Absolute path to the AIT static/ directory
        """
        super().__init__(ait_static_files_path)
        self._target_type = 1
        self._template_path = template_path
        self._ait_include_path = ait_static_files_path
        self._arch = arch
        self._kwargs = kwargs
        self._compile_options = self._build_compile_options()
        if cuda_version is None:
            # try to set default CUDA version based on the arch
            if arch == "80":
                cuda_version = "11.4.2"
            elif arch == "90":
                cuda_version = "12.0.0"
        self._cuda_version = cuda_version
CUDA target init.

Parameters
----------
template_path : str, optional
by default "${repo_root}/3rdparty/cutlass"
ait_static_files_path : str
Absolute path to the AIT static/ directory
python/aitemplate/backend/cuda/target_def.py