Method _validate_languages

src/sol_execbench/core/data/solution.py:175–231

Validate languages support matrix. Raises ------ ValueError If the languages are not valid.

(self)

    @model_validator(mode="after")
    def _validate_languages(self) -> "BuildSpec":
        """Validate languages support matrix.

        Raises
        ------
        ValueError
            If the languages are not valid.
        """

        python_languages = [
            SupportedLanguages.PYTORCH,
            SupportedLanguages.TRITON,
            SupportedLanguages.CUTE_DSL,
            SupportedLanguages.CUTILE,
            SupportedLanguages.CUDNN_FRONTEND,
        ]
        cpp_languages = [
            SupportedLanguages.CUTLASS,
            SupportedLanguages.CUDNN,
            SupportedLanguages.CUBLAS,
            SupportedLanguages.CUDA_CPP,
        ]

        included_python_langs = [
            language for language in self.languages if language in python_languages
        ]
        included_cpp_langs = [
            language for language in self.languages if language in cpp_languages
        ]
        if len(included_cpp_langs) and len(included_python_langs):
            raise ValueError(
                f"C++ and Python cannot be mixed, but got {included_cpp_langs} and {included_python_langs}"
            )

        # Validate entry point file suffix matches the language category.
        entry_file = self.entry_point.split("::")[0]
        suffix = Path(entry_file).suffix
        if included_cpp_langs and suffix not in (
            ".cu",
            ".cpp",
            ".cc",
            ".cxx",
            ".c",
            ".h",
            ".hpp",
            ".cuh",
        ):
            raise ValueError(
                f"C++ languages require a C++/CUDA entry point file, "
                f"but got '{entry_file}' (suffix '{suffix}')"
            )
        if included_python_langs and suffix != ".py":
            raise ValueError(
                f"Python languages require a .py entry point file, "
                f"but got '{entry_file}' (suffix '{suffix}')"
            )
        return self