sleipnirgroup-jormungandr 0.5.1.dev5__tar.gz → 0.5.1.dev6__tar.gz

{sleipnirgroup_jormungandr-0.5.1.dev5 → sleipnirgroup_jormungandr-0.5.1.dev6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sleipnirgroup-jormungandr
-Version: 0.5.1.dev5
+Version: 0.5.1.dev6
 Summary: Reverse mode autodiff library and NLP solver DSL
 License-Expression: BSD-3-Clause
 License-File: LICENSE.txt

{sleipnirgroup_jormungandr-0.5.1.dev5 → sleipnirgroup_jormungandr-0.5.1.dev6}/include/sleipnir/optimization/problem.hpp

@@ -351,6 +351,10 @@ class Problem {
 
       ad_setup_profilers[0].stop();
 
+      if (options.diagnostics) {
+        print_setup_diagnostics(ad_setup_profilers);
+      }
+
 #ifndef SLEIPNIR_DISABLE_DIAGNOSTICS
       // Sparsity pattern files written when spy flag is set
       std::unique_ptr<Spy<Scalar>> H_spy;
@@ -413,6 +417,10 @@ class Problem {
 
       ad_setup_profilers[0].stop();
 
+      if (options.diagnostics) {
+        print_setup_diagnostics(ad_setup_profilers);
+      }
+
 #ifndef SLEIPNIR_DISABLE_DIAGNOSTICS
       // Sparsity pattern files written when spy flag is set
       std::unique_ptr<Spy<Scalar>> H_spy;
@@ -503,6 +511,10 @@ class Problem {
 
       ad_setup_profilers[0].stop();
 
+      if (options.diagnostics) {
+        print_setup_diagnostics(ad_setup_profilers);
+      }
+
 #ifndef SLEIPNIR_DISABLE_DIAGNOSTICS
       // Sparsity pattern files written when spy flag is set
       std::unique_ptr<Spy<Scalar>> H_spy;
@@ -599,7 +611,6 @@ class Problem {
     }
 
     if (options.diagnostics) {
-      print_autodiff_diagnostics(ad_setup_profilers);
       slp::println("\nExit: {}", status);
     }
 

{sleipnirgroup_jormungandr-0.5.1.dev5 → sleipnirgroup_jormungandr-0.5.1.dev6}/include/sleipnir/util/print_diagnostics.hpp

@@ -318,17 +318,17 @@ inline void print_solver_diagnostics(
 #endif
 
 #ifndef SLEIPNIR_DISABLE_DIAGNOSTICS
-/// Prints autodiff diagnostics.
+/// Prints setup diagnostics.
 ///
-/// @param setup_profilers Autodiff setup profilers.
-inline void print_autodiff_diagnostics(
+/// @param setup_profilers Setup profilers.
+inline void print_setup_diagnostics(
     const gch::small_vector<SetupProfiler>& setup_profilers) {
   auto setup_duration = to_ms(setup_profilers[0].duration());
 
   // Print heading
   slp::println("┏{:━^21}┯{:━^18}┯{:━^10}┯{:━^9}┯{:━^4}┓", "", "", "", "", "");
-  slp::println("┃{:^21}│{:^18}│{:^10}│{:^9}│{:^4}┃", "autodiff trace",
-               "percent", "total (ms)", "each (ms)", "runs");
+  slp::println("┃{:^21}│{:^18}│{:^10}│{:^9}│{:^4}┃", "setup trace", "percent",
+               "total (ms)", "each (ms)", "runs");
   slp::println("┡{:━^21}┷{:━^18}┷{:━^10}┷{:━^9}┷{:━^4}┩", "", "", "", "", "");
 
   // Print setup profilers
@@ -344,7 +344,7 @@ inline void print_autodiff_diagnostics(
   slp::println("└{:─^66}┘", "");
 }
 #else
-#define print_autodiff_diagnostics(...)
+#define print_setup_diagnostics(...)
 #endif
 
 }  // namespace slp

{sleipnirgroup_jormungandr-0.5.1.dev5 → sleipnirgroup_jormungandr-0.5.1.dev6}/python/src/sleipnir/__init__.py

@@ -2,4 +2,4 @@
 
 from ._sleipnir import *
 
-__version__ = "0.5.1.dev5"
+__version__ = "0.5.1.dev6"