experimental/dispatch_profiler/generator.py

# Copyright 2023 The IREE Authors
#
# Licensed under the Apache License v2.0 with LLVM Exceptions.
# See https://llvm.org/LICENSE.txt for license information.
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

import argparse
from library import *
from matmul import *
from manifest import *
from options import parse_generator_arguments

###############################################################################

if __name__ == "__main__":

  parser = argparse.ArgumentParser(description="Generates MLIR operations for "\
                     "verification and profiling of IREE compiled dispatches.")

  args = parse_generator_arguments(parser)

  # Manifest dispatches for a group of accompanying operations and configurations.
  manifest = Manifest(args)

  # Load all the pre-defined dispatches in a manifest.
  manifest.initialize()

  # Emit the dispatches in MLIR source files.
  manifest.emit()