hespas.chakra_gen.chakra_config

Classes

ChakraGenConfig(input[, output_dir, ...])

class hespas.chakra_gen.chakra_config.ChakraGenConfig(input, output_dir=None, mlir_file=None, clean=None, disable_cache=None, cache_dir=None, num_threads=-1, num_npus=None, split_choice=None, split_kwargs=None)

Bases: object

__init__(input, output_dir=None, mlir_file=None, clean=None, disable_cache=None, cache_dir=None, num_threads=-1, num_npus=None, split_choice=None, split_kwargs=None)