enable tf serving gpu test (kubeflow#497)

jlewi · Mar 28, 2018 · c0f9b49 · c0f9b49
1 parent 5ca6cc7
commit c0f9b49
Showing 1 changed file with 10 additions and 11 deletions.
diff --git a/components/k8s-model-server/images/releaser/components/workflows.libsonnet b/components/k8s-model-server/images/releaser/components/workflows.libsonnet
@@ -232,17 +232,16 @@
           template: "test-tf-serving",
           dependencies: ["deploy-tf-serving"],
         },
-        // TODO(lunkai): investigate gpu flakiness, then uncomment.
-        // {
-        //   name: "deploy-tf-serving-gpu",
-        //   template: "deploy-tf-serving-gpu",
-        //   dependencies: ["checkout"],
-        // },
-        // {
-        //   name: "test-tf-serving-gpu",
-        //   template: "test-tf-serving-gpu",
-        //   dependencies: ["deploy-tf-serving-gpu"],
-        // },
+        {
+          name: "deploy-tf-serving-gpu",
+          template: "deploy-tf-serving-gpu",
+          dependencies: ["checkout"],
+        },
+        {
+          name: "test-tf-serving-gpu",
+          template: "test-tf-serving-gpu",
+          dependencies: ["deploy-tf-serving-gpu"],
+        },
       ];
       local e2e_tasks = e2e_tasks_base + if build_image then [
         {