Google Git
Sign in
opensecura / 3p / openxla / iree / f1adc7897c620fd835da718c68ce0c4156d45688 / . / build_tools / python / e2e_test_framework / models
tree: 4a0426fd79cb342c2d1ce2c6305a64ac462fb331 [path history] [tgz]
  1. __init__.py
  2. CMakeLists.txt
  3. jax_models.py
  4. matmul.py
  5. model_groups.py
  6. README.md
  7. tf_models.py
  8. tflite_models.py
  9. torch_models.py
  10. utils.py
  11. utils_test.py
build_tools/python/e2e_test_framework/models/README.md

IREE E2E Test Artifacts Models

This directory contains the definitions of source models for e2e tests (including the benchmark suites). Each source type of models is defined in <source_type>_models.py

Adding a new model

  1. Upload the source model file to the GCS bucket gs://iree-model-artifacts.
    • You can ask IREE team members for help if you don't have access.
  2. Register a unique model ID in build_tools/python/e2e_test_framework/unique_ids.py.
    • Run uuid.uuid4(), or uuidgen on Linux, to generate a new uuid for the new model. The ID is a totally arbitrary artificial key to distinguish different models.
  3. Define a new model with GCS URL and model ID in <source_type>_models.py.
  4. Optionally add the model to a model group in build_tools/python/e2e_test_framework/models/model_groups.py.