Add posttestspec and release_buildspec (#158)

aws · May 9, 2020 · 66d0fe1 · 66d0fe1
1 parent fa21368
commit 66d0fe1
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 0 deletions.
diff --git a/posttestspec.yml b/posttestspec.yml
@@ -0,0 +1,6 @@
+version: 0.2
+
+phases:
+  build:
+    commands:
+      - mark_images_as_tested --images-file $CODEBUILD_SRC_DIR_DLC_IMAGES_JSON/test_type_images.json
diff --git a/release_buildspec.yml b/release_buildspec.yml
@@ -0,0 +1,9 @@
+version: 0.2
+
+phases:
+  pre_build:
+    commands:
+      - start-dockerd
+  build:
+    commands:
+      - publish_dlc_images --release-spec $(pwd)/release_images.yml
diff --git a/release_images.yml b/release_images.yml
@@ -0,0 +1,17 @@
+---
+release_images:
+  1:
+    framework: "pytorch"
+    version: "1.5.0"
+    training:
+      device_types: ["cpu", "gpu"]
+      python_versions: ["py36"]
+      os_version: "ubuntu16.04"
+      cuda_version: "cu101"
+      dlc_version: "1.1"
+    inference:
+      device_types: ["cpu", "gpu"]
+      python_versions: ["py36"]
+      os_version: "ubuntu16.04"
+      cuda_version: "cu101"
+      dlc_version: "1.1"