Add a CI job for cpp/dcgan

pytorch · Jan 23, 2024 · 0b78d50 · 0b78d50
1 parent d78cff0
commit 0b78d50
Showing 1 changed file with 20 additions and 4 deletions.
diff --git a/run_cpp_examples.sh b/run_cpp_examples.sh
@@ -94,6 +94,21 @@ function custom-dataset() {
     exit 1
   fi
 }
+function dcgan() {
+  start
+  mkdir build
+  cd build
+  cmake -DCMAKE_PREFIX_PATH=$LIBTORCH_PATH ..
+  make
+  if [ $? -eq 0 ]; then
+    echo "Successfully built $EXAMPLE"
+    ./$EXAMPLE # Run the executable
+    check_run_success $EXAMPLE
+  else
+    error "Failed to build $EXAMPLE"
+    exit 1
+  fi
+}
 
 function mnist() {
   start
@@ -139,10 +154,11 @@ function clean() {
 }
 
 function run_all() {
-  autograd
-  custom-dataset
-  mnist
-  regression
+  dcgan
+  # autograd
+  # custom-dataset
+  # mnist
+  # regression
 }
 
 # by default, run all examples