From a6e5c136983a2b12104e2c3ba86a96e7e7e63f60 Mon Sep 17 00:00:00 2001
From: Alexey Panferov <lexavtanke@gmail.com>
Date: Wed, 18 Oct 2023 16:38:30 +0300
Subject: [PATCH] feat(ansible): add tvm artifacts download to artifacts role

Signed-off-by: Alexey Panferov <lexavtanke@gmail.com>
---
 ansible/roles/artifacts/tasks/main.yaml | 93 +++++++++++++++++++++++++
 1 file changed, 93 insertions(+)

diff --git a/ansible/roles/artifacts/tasks/main.yaml b/ansible/roles/artifacts/tasks/main.yaml
index 4be7a3400c9..1628d3eca80 100644
--- a/ansible/roles/artifacts/tasks/main.yaml
+++ b/ansible/roles/artifacts/tasks/main.yaml
@@ -361,3 +361,96 @@
     dest: "{{ data_dir }}/traffic_light_ssd_fine_detector/voc_labels_tl.txt"
     mode: "644"
     checksum: sha256:a41e6e3324e32c30b3b2fe38908eaf3471e2bfdaeb9e14ca0c1c3bc0275119c6
+
+# tvm_utility 
+- name: Create tvm_utility/models directory inside {{ data_dir }}
+  ansible.builtin.file:
+    path: "{{ data_dir }}/tvm_utility/models"
+    mode: "755"
+    state: directory
+
+- name: Download yolo_v2_tiny-x86_64-llvm-3.0.0-20221221.tar.gz
+  become: true
+  ansible.builtin.get_url:
+    url: https://autoware-modelzoo.s3.us-east-2.amazonaws.com/models/3.0.0-20221221/yolo_v2_tiny-x86_64-llvm-3.0.0-20221221.tar.gz
+    dest: "{{ data_dir }}/tvm_utility/yolo_v2_tiny-x86_64-llvm-3.0.0-20221221.tar.gz"
+    checksum: sha256:66b3ca668e577393b657fbe1ed626538d89ca3adccd5862de6c7fa190238dbca
+
+- name: Create yolo_v2_tiny folder in tvm_utility/models of {{ data_dir }}
+  ansible.builtin.file:
+    path: "{{ data_dir }}/tvm_utility/models/yolo_v2_tiny"
+    mode: "755"
+    state: directory
+
+- name: Extract yolo_v2_tiny-x86_64-llvm-3.0.0-20221221.tar.gz
+  ansible.builtin.unarchive:
+    src: "{{ data_dir }}/tvm_utility/yolo_v2_tiny-x86_64-llvm-3.0.0-20221221.tar.gz"
+    dest: "{{ data_dir }}/tvm_utility/models/yolo_v2_tiny"
+
+# lidar_centerpoint_tvm
+- name: Create lidar_centerpoint_tvm/models directory inside {{ data_dir }}
+  ansible.builtin.file:
+    path: "{{ data_dir }}/lidar_centerpoint_tvm/models"
+    mode: "755"
+    state: directory
+
+- name: Download centerpoint_encoder-x86_64-llvm-3.0.0-20221221.tar.gz
+  become: true
+  ansible.builtin.get_url:
+    url: https://autoware-modelzoo.s3.us-east-2.amazonaws.com/models/3.0.0-20221221/centerpoint_encoder-x86_64-llvm-3.0.0-20221221.tar.gz
+    dest: "{{ data_dir }}/lidar_centerpoint_tvm/centerpoint_encoder-x86_64-llvm-3.0.0-20221221.tar.gz"
+    checksum: sha256:41e19de43f30549c325968aee06b4ad0d9701220be819e79d37efdfa86b918d0
+
+- name: Create centerpoint_encoder folder in lidar_centerpoint_tvm/models of {{ data_dir }}
+  ansible.builtin.file:
+    path: "{{ data_dir }}/lidar_centerpoint_tvm/models/centerpoint_encoder"
+    mode: "755"
+    state: directory
+
+- name: Extract centerpoint_encoder-x86_64-llvm-3.0.0-20221221.tar.gz
+  ansible.builtin.unarchive:
+    src: "{{ data_dir }}/lidar_centerpoint_tvm/centerpoint_encoder-x86_64-llvm-3.0.0-20221221.tar.gz"
+    dest: "{{ data_dir }}/lidar_centerpoint_tvm/models/centerpoint_encoder"
+
+- name: Download centerpoint_backbone-x86_64-llvm-3.0.0-20221221.tar.gz
+  become: true
+  ansible.builtin.get_url:
+    url: https://autoware-modelzoo.s3.us-east-2.amazonaws.com/models/3.0.0-20221221/centerpoint_backbone-x86_64-llvm-3.0.0-20221221.tar.gz
+    dest: "{{ data_dir }}/lidar_centerpoint_tvm/centerpoint_backbone-x86_64-llvm-3.0.0-20221221.tar.gz"
+    checksum: sha256:3840b6b3590984e8115d66b12061aea3a2cfaed70b4e8d59457f04b0d6f6a1fc
+
+- name: Create centerpoint_backbone folder in lidar_centerpoint_tvm/models of {{ data_dir }}
+  ansible.builtin.file:
+    path: "{{ data_dir }}/lidar_centerpoint_tvm/models/centerpoint_backbone"
+    mode: "755"
+    state: directory
+
+- name: Extract centerpoint_backbone-x86_64-llvm-3.0.0-20221221.tar.gz
+  ansible.builtin.unarchive:
+    src: "{{ data_dir }}/lidar_centerpoint_tvm/centerpoint_backbone-x86_64-llvm-3.0.0-20221221.tar.gz"
+    dest: "{{ data_dir }}/lidar_centerpoint_tvm/models/centerpoint_backbone"
+
+# lidar_apollo_segmentation_tvm
+- name: Create lidar_apollo_segmentation_tvm/models directory inside {{ data_dir }}
+  ansible.builtin.file:
+    path: "{{ data_dir }}/lidar_apollo_segmentation_tvm/models"
+    mode: "755"
+    state: directory
+
+- name: Download baidu_cnn-x86_64-llvm-3.0.0-20221221.tar.gz
+  become: true
+  ansible.builtin.get_url:
+    url: https://autoware-modelzoo.s3.us-east-2.amazonaws.com/models/3.0.0-20221221/baidu_cnn-x86_64-llvm-3.0.0-20221221.tar.gz
+    dest: "{{ data_dir }}/lidar_apollo_segmentation_tvm/baidu_cnn-x86_64-llvm-3.0.0-20221221.tar.gz"
+    checksum: sha256:4293e6196ec937d2cd5ec658e5ce70933647d2d38633a1805febb36cafd684e3
+
+- name: Create baidu_cnn folder in lidar_apollo_segmentation_tvm/models of {{ data_dir }}
+  ansible.builtin.file:
+    path: "{{ data_dir }}/lidar_apollo_segmentation_tvm/models/baidu_cnn"
+    mode: "755"
+    state: directory
+
+- name: Extract baidu_cnn-x86_64-llvm-3.0.0-20221221.tar.gz
+  ansible.builtin.unarchive:
+    src: "{{ data_dir }}/lidar_apollo_segmentation_tvm/baidu_cnn-x86_64-llvm-3.0.0-20221221.tar.gz"
+    dest: "{{ data_dir }}/lidar_apollo_segmentation_tvm/models/baidu_cnn"
\ No newline at end of file