Source code for compiler_gym.envs.gcc.datasets

# Copyright (c) Facebook, Inc. and its affiliates.
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.
from functools import lru_cache
from pathlib import Path
from typing import Iterable, List, Optional, Union

from compiler_gym.datasets import Dataset
from compiler_gym.envs.gcc.datasets.anghabench import AnghaBenchDataset
from compiler_gym.envs.gcc.datasets.chstone import CHStoneDataset
from compiler_gym.envs.gcc.datasets.csmith import CsmithBenchmark, CsmithDataset
from compiler_gym.util.runfiles_path import site_data_path

def _get_gcc_datasets(
    gcc_bin: Union[str, Path], site_data_base: Optional[Path] = None
) -> Iterable[Dataset]:
    site_data_base = site_data_base or site_data_path("gcc-v0")

    yield CHStoneDataset(gcc_bin=gcc_bin, site_data_base=site_data_base)
    yield AnghaBenchDataset(site_data_base=site_data_base)
    yield CsmithDataset(gcc_bin=gcc_bin, site_data_base=site_data_base)

[docs]@lru_cache(maxsize=16) def get_gcc_datasets( gcc_bin: Union[str, Path], site_data_base: Optional[Path] = None ) -> List[Dataset]: """Instantiate the builtin GCC datasets. :param gcc_bin: The GCC binary to use. :param site_data_base: The root of the site data path. :return: An iterable sequence of :class:`Dataset <compiler_gym.datasets.Dataset>` instances. """ return list(_get_gcc_datasets(gcc_bin, site_data_base))
__all__ = [ "AnghaBenchDataset", "CHStoneDataset", "CsmithBenchmark", "CsmithDataset", "get_gcc_datasets", ]