Add test case check

LLNL · Jun 4, 2024 · 9c1f616 · 9c1f616
1 parent 8292c53
commit 9c1f616
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/AMSlib/torch_model.cpp b/tests/AMSlib/torch_model.cpp
@@ -74,6 +74,9 @@ int main(int argc, char *argv[])
         SurrogateModel<float>::getInstance(model_path, resource);
     assert(!model->is_double());
     inference<float>(*model, resource);
+  } else {
+    std::cout << "Unknown data type " << data_type << "\n";
+    return 1;
   }
 
   return 0;