sleipnirgroup-jormungandr 0.6.6.dev10__tar.gz → 0.6.6.dev12__tar.gz

{sleipnirgroup_jormungandr-0.6.6.dev10 → sleipnirgroup_jormungandr-0.6.6.dev12}/CMakeLists.txt

@@ -357,8 +357,7 @@ if(SLEIPNIR_BUILD_PYTHON)
         COMPONENT python_modules
     )
 
-    # pybind11_mkdoc doesn't support Windows
-    if(NOT WIN32 AND NOT CMAKE_CROSSCOMPILING)
+    if(NOT CMAKE_CROSSCOMPILING)
         # pybind11_mkdoc dependency
         FetchContent_Declare(
             pybind11_mkdoc

{sleipnirgroup_jormungandr-0.6.6.dev10 → sleipnirgroup_jormungandr-0.6.6.dev12}/PKG-INFO

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

{sleipnirgroup_jormungandr-0.6.6.dev10 → sleipnirgroup_jormungandr-0.6.6.dev12}/cmake/Pybind11Mkdoc.cmake

@@ -1,12 +1,9 @@
 function(pybind11_mkdoc target headers)
     find_package(Python3 REQUIRED COMPONENTS Interpreter)
 
-    get_target_property(target_dirs ${target} INCLUDE_DIRECTORIES)
-    list(TRANSFORM target_dirs PREPEND "-I")
-
-    get_target_property(eigen_dirs Eigen3::Eigen INTERFACE_INCLUDE_DIRECTORIES)
-    list(FILTER eigen_dirs INCLUDE REGEX "\\$<BUILD_INTERFACE:.*>")
-    list(TRANSFORM eigen_dirs PREPEND "-I")
+    # Target compiler flags
+    get_target_property(target_flags ${target} INCLUDE_DIRECTORIES)
+    list(TRANSFORM target_flags PREPEND "-I")
 
     # Get default clang version
     execute_process(
@@ -16,6 +13,7 @@ function(pybind11_mkdoc target headers)
     )
     string(REGEX MATCH "[0-9]+" clang_version ${clang_version})
 
+    # LLVM environment variables
     if(UNIX AND NOT APPLE)
         if(EXISTS /usr/lib/libclang.so)
             set(env_vars
@@ -28,26 +26,46 @@ function(pybind11_mkdoc target headers)
                 LIBCLANG_PATH=/usr/lib/llvm-${clang_version}/lib/libclang.so
             )
         endif()
-        set(llvm_dirs -I/usr/lib/clang/${clang_version}/include)
     endif()
 
+    # Clang compiler flags
+    if(UNIX AND NOT APPLE)
+        set(clang_flags -I/usr/lib/clang/${clang_version}/include)
+    endif()
+
+    # Eigen compiler flags
+    get_target_property(eigen_flags Eigen3::Eigen INTERFACE_INCLUDE_DIRECTORIES)
+    list(FILTER eigen_flags INCLUDE REGEX "\\$<BUILD_INTERFACE:.*>")
+    list(TRANSFORM eigen_flags PREPEND "-I")
+
+    # clang on Windows aarch64 failed to parse Eigen's NEON headers
+    list(APPEND eigen_flags "-DEIGEN_DONT_VECTORIZE")
+
+    # small_vector compiler flags
     get_target_property(
-        small_vector_dirs
+        small_vector_flags
         small_vector
         INTERFACE_INCLUDE_DIRECTORIES
     )
-    list(FILTER small_vector_dirs INCLUDE REGEX "\\$<BUILD_INTERFACE:.*>")
-    list(TRANSFORM small_vector_dirs PREPEND "-I")
+    list(FILTER small_vector_flags INCLUDE REGEX "\\$<BUILD_INTERFACE:.*>")
+    list(TRANSFORM small_vector_flags PREPEND "-I")
 
+    # TODO: Remove when Python 3.15 makes UTF-8 the default
+    set(env_vars ${env_vars} PYTHONUTF8=1)
+
+    # Generate docstrings.hpp
     add_custom_command(
         OUTPUT ${CMAKE_CURRENT_SOURCE_DIR}/python/cpp/docstrings.hpp
         COMMAND
             ${CMAKE_COMMAND} -E env ${env_vars} ${Python3_EXECUTABLE} -m
             pybind11_mkdoc ${headers} -o
-            ${CMAKE_CURRENT_SOURCE_DIR}/python/cpp/docstrings.hpp ${target_dirs}
-            ${eigen_dirs} ${llvm_dirs} ${small_vector_dirs} -std=c++23
+            ${CMAKE_CURRENT_SOURCE_DIR}/python/cpp/docstrings.hpp
+            ${target_flags} ${clang_flags} ${eigen_flags} ${small_vector_flags}
+            -std=c++23
         DEPENDS ${headers}
     )
+
+    # Add docstrings target
     add_custom_target(
         ${target}_docstrings
         DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/python/cpp/docstrings.hpp

{sleipnirgroup_jormungandr-0.6.6.dev10 → sleipnirgroup_jormungandr-0.6.6.dev12}/include/sleipnir/autodiff/variable_matrix.hpp

@@ -1110,10 +1110,17 @@ class VariableMatrix : public SleipnirBase {
         Eigen::Vector<Scalar, Eigen::Dynamic>::Constant(rows(), q[0])
             .asDiagonal()};
     auto A_pow = *this;
-    for (size_t k = 1; k < NUM_COEFFS; ++k) {
+    size_t k = 1;
+    while (true) {
       P += p[k] * A_pow;
       Q += q[k] * A_pow;
-      A_pow *= *this;
+
+      ++k;
+      if (k < NUM_COEFFS) {
+        A_pow *= *this;
+      } else {
+        break;
+      }
     }
 
     // https://mpmath.org/doc/current/calculus/approximation.html#mpmath.pade

{sleipnirgroup_jormungandr-0.6.6.dev10 → sleipnirgroup_jormungandr-0.6.6.dev12}/python/src/sleipnir/__init__.py

@@ -2,4 +2,4 @@
 
 from ._sleipnir import *
 
-__version__ = "0.6.6.dev10"
+__version__ = "0.6.6.dev12"