| #!/usr/bin/env python3 |
| # Licensed to the Apache Software Foundation (ASF) under one |
| # or more contributor license agreements. See the NOTICE file |
| # distributed with this work for additional information |
| # regarding copyright ownership. The ASF licenses this file |
| # to you under the Apache License, Version 2.0 (the |
| # "License"); you may not use this file except in compliance |
| # with the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, |
| # software distributed under the License is distributed on an |
| # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| # KIND, either express or implied. See the License for the |
| # specific language governing permissions and limitations |
| # under the License. |
| import sys |
| |
| |
| files_to_stash = { |
| # Executables and build files needed to run c++ tests |
| "cpptest": ["build/cpptest", "build/build.ninja", "build/CMakeFiles/rules.ninja"], |
| # Executables and build files needed to c runtime tests |
| "crttest": ["build/crttest"], |
| # Folder for hexagon build |
| "hexagon_api": [ |
| "build/hexagon_api_output", |
| ], |
| # Folder for microtvm build |
| "microtvm_template_projects": [ |
| "build/microtvm_template_projects", |
| ], |
| # Folders and build files for c runtime |
| "standalone_crt": ["build/standalone_crt", "build/build.ninja"], |
| # This library is produced with HIDE_PRIVATE_SYMBOLS=ON |
| "tvm_allvisible": ["build/libtvm_allvisible.so"], |
| # runtime files |
| "tvm_runtime": ["build/libtvm_runtime.so", "build/config.cmake"], |
| # compiler files |
| "tvm_lib": ["build/libtvm.so", "build/libtvm_runtime.so", "build/config.cmake"], |
| # compiler files and fsim |
| "tvm_multilib": [ |
| "build/libtvm.so", |
| "build/libvta_fsim.so", |
| "build/libtvm_runtime.so", |
| "build/config.cmake", |
| ], |
| # compiler files, fsim, and tsim |
| "tvm_multilib_tsim": [ |
| "build/libvta_tsim.so", |
| "build/libtvm.so", |
| "build/libvta_fsim.so", |
| "build/libtvm_runtime.so", |
| "build/config.cmake", |
| ], |
| } |
| |
| |
| # AWS info |
| aws_default_region = "us-west-2" |
| aws_ecr_url = "dkr.ecr." + aws_default_region + ".amazonaws.com" |
| |
| # Docker Images |
| docker_images = { |
| "ci_arm": { |
| "tag": "tlcpack/ci-arm:20221013-060115-61c9742ea", |
| "platform": "ARM", |
| }, |
| "ci_cortexm": { |
| "tag": "tlcpack/ci-cortexm:20221013-060115-61c9742ea", |
| "platform": "CPU", |
| }, |
| "ci_cpu": { |
| "tag": "tlcpack/ci-cpu:20221013-060115-61c9742ea", |
| "platform": "CPU", |
| }, |
| "ci_gpu": { |
| "tag": "tlcpack/ci-gpu:20221019-060125-0b4836739", |
| "platform": "GPU", |
| }, |
| "ci_hexagon": { |
| "tag": "tlcpack/ci-hexagon:20221013-060115-61c9742ea", |
| "platform": "CPU", |
| }, |
| "ci_i386": { |
| "tag": "tlcpack/ci-i386:20221013-060115-61c9742ea", |
| "platform": "CPU", |
| }, |
| "ci_lint": { |
| "tag": "tlcpack/ci-lint:20221013-060115-61c9742ea", |
| "platform": "CPU", |
| }, |
| "ci_minimal": { |
| "tag": "tlcpack/ci-minimal:20221013-060115-61c9742ea", |
| "platform": "CPU", |
| }, |
| "ci_riscv": { |
| "tag": "tlcpack/ci-riscv:20221013-060115-61c9742ea", |
| "platform": "CPU", |
| }, |
| "ci_wasm": { |
| "tag": "tlcpack/ci-wasm:20221013-060115-61c9742ea", |
| "platform": "CPU", |
| }, |
| } |
| |
| data = { |
| "images": [{"name": k, "platform": v["platform"]} for k, v in docker_images.items()], |
| "aws_default_region": aws_default_region, |
| "aws_ecr_url": aws_ecr_url, |
| **{k: v["tag"] for k, v in docker_images.items()}, |
| **files_to_stash, |
| } |
| |
| if __name__ == "__main__": |
| # This is used in docker/dev_common.sh to look up image tags |
| name = sys.argv[1] |
| if name in docker_images: |
| print(docker_images[name]["tag"]) |
| else: |
| exit(1) |