forward pass

Author: dance858

.clang-format

@@ -0,0 +1,45 @@
+# .clang-format for C project with Allman style braces
+BasedOnStyle: LLVM
+
+# Indentation
+IndentWidth: 4
+UseTab: Never
+TabWidth: 4
+
+# Line length
+ColumnLimit: 85
+BinPackParameters: true
+BinPackArguments: true
+
+# Braces style
+BreakBeforeBraces: Allman   # Allman style: brace on next line
+BraceWrapping:
+  AfterFunction: true
+  AfterControlStatement: true
+  AfterStruct: true
+  AfterClass: true
+  AfterNamespace: false
+  IndentBraces: false
+
+# Pointers
+PointerAlignment: Right      
+
+# Spaces
+SpaceAfterCStyleCast: true
+SpaceAfterLogicalNot: false
+SpacesInParentheses: false
+SpacesInContainerLiterals: false
+SpaceBeforeParens: ControlStatements
+
+# Allow single-line short statements if simple
+AllowShortIfStatementsOnASingleLine: true
+AllowShortLoopsOnASingleLine: true
+AllowShortFunctionsOnASingleLine: Inline
+AllowAllParametersOfDeclarationOnNextLine: true
+
+
+# Other
+ReflowComments: true
+IndentCaseLabels: true
+AlignOperands: true
+InsertNewlineAtEOF: true              

CMakeLists.txt

@@ -0,0 +1,37 @@
+cmake_minimum_required(VERSION 3.10)
+project(DNLP_Diff_Engine C)
+
+set(CMAKE_C_STANDARD 99)
+
+# Include directories
+include_directories(${PROJECT_SOURCE_DIR}/include)
+
+# Source files
+set(SOURCES
+    src/expr.c
+    src/elementwise/exp.c
+    src/elementwise/log.c
+    src/affine/variable.c
+    src/affine/constant.c
+    src/affine/add.c
+)
+
+# Create library
+add_library(dnlp_diff ${SOURCES})
+target_link_libraries(dnlp_diff m)
+
+# Enable testing
+enable_testing()
+
+# Single test executable that links all test source files
+add_executable(test_all
+    tests/forward_pass/all_tests.c
+    tests/forward_pass/test_helpers.c
+    tests/forward_pass/affine/test_variable_constant.c
+    tests/forward_pass/affine/test_add.c
+    tests/forward_pass/elementwise/test_exp.c
+    tests/forward_pass/elementwise/test_log.c
+    tests/forward_pass/composite/test_composite.c
+)
+target_link_libraries(test_all dnlp_diff)
+add_test(NAME AllTests COMMAND test_all)

include/affine/add.h

@@ -0,0 +1,8 @@
+#ifndef ADD_H
+#define ADD_H
+
+#include "expr.h"
+
+expr *new_add(expr *left, expr *right);
+
+#endif /* ADD_H */

include/affine/constant.h

@@ -0,0 +1,8 @@
+#ifndef CONSTANT_H
+#define CONSTANT_H
+
+#include "expr.h"
+
+expr *new_constant(int m, const double *values);
+
+#endif /* CONSTANT_H */

include/affine/variable.h

@@ -0,0 +1,8 @@
+#ifndef VARIABLE_H
+#define VARIABLE_H
+
+#include "expr.h"
+
+expr *new_variable(int m);
+
+#endif /* VARIABLE_H */

include/elementwise.h

@@ -0,0 +1,9 @@
+#ifndef ELEMENTWISE_H
+#define ELEMENTWISE_H
+
+#include "expr.h"
+
+expr *new_exp(expr *child);
+expr *new_log(expr *child);
+
+#endif /* ELEMENTWISE_H */

include/expr.h

@@ -0,0 +1,26 @@
+#ifndef EXPR_H
+#define EXPR_H
+
+#include <stddef.h>
+
+/* Forward declaration */
+struct expr;
+
+/* Function pointer type for forward pass computation */
+typedef void (*forward_fn)(struct expr *node, const double *u);
+
+/* Expression node structure */
+typedef struct expr
+{
+    int m;              /* Output dimension */
+    double *value;      /* Preallocated output value array */
+    struct expr *left;  /* Left child (can be NULL) */
+    struct expr *right; /* Right child (can be NULL) */
+    forward_fn forward; /* Forward pass function */
+} expr;
+
+/* Memory management functions */
+expr *new_expr(int m);
+void free_expr(expr *node);
+
+#endif /* EXPR_H */

src/affine/add.c

@@ -0,0 +1,29 @@
+#include "affine/add.h"
+
+static void add_forward(expr *node, const double *u)
+{
+    /* children's forward passes */
+    node->left->forward(node->left, u);
+    node->right->forward(node->right, u);
+
+    /* add left and right values */
+    for (int i = 0; i < node->m; i++)
+    {
+        node->value[i] = node->left->value[i] + node->right->value[i];
+    }
+}
+
+expr *new_add(expr *left, expr *right)
+{
+    if (!left || !right) return NULL;
+    if (left->m != right->m) return NULL; /* Dimension mismatch */
+
+    expr *node = new_expr(left->m);
+    if (!node) return NULL;
+
+    node->left = left;
+    node->right = right;
+    node->forward = add_forward;
+
+    return node;
+}

src/affine/constant.c

@@ -0,0 +1,22 @@
+#include "affine/constant.h"
+#include <string.h>
+
+static void constant_forward(expr *node, const double *u)
+{
+    /* Constants don't depend on u; values are already set */
+    (void) node;
+    (void) u;
+}
+
+expr *new_constant(int m, const double *values)
+{
+    expr *node = new_expr(m);
+    if (!node) return NULL;
+
+    /* Copy constant values */
+    memcpy(node->value, values, m * sizeof(double));
+
+    node->forward = constant_forward;
+
+    return node;
+}

src/affine/variable.c

@@ -0,0 +1,17 @@
+#include "affine/variable.h"
+#include <string.h>
+
+static void variable_forward(expr *node, const double *u)
+{
+    memcpy(node->value, u, node->m * sizeof(double));
+}
+
+expr *new_variable(int m)
+{
+    expr *node = new_expr(m);
+    if (!node) return NULL;
+
+    node->forward = variable_forward;
+
+    return node;
+}