feat: scaffold FCES-native C++ project with libtorch integration

- CMakeLists.txt with libtorch, GoogleTest, GoogleBenchmark, OpenMP, pybind11
- Header files: config, controller, population, fitness, evolution, spectral, oscillation, telemetry, optimizer
- Source implementations: controller (full micro-MLP forward pass, mutation, crossover), fitness (Welford's algorithm), oscillation (DFT), spectral (SVD rank), optimizer (sign-SGD stub)
- Tests: controller, population, fitness, optimizer (Google Test)
- Benchmarks: evolve throughput, optimizer step (Google Benchmark)
- Examples: simple optimization, PyTorch/libtorch integration
- Python extension: pybind11 bindings with setup.py
- README with architecture diagram and build instructions
This commit is contained in:
AI-anonymous
2026-05-19 16:05:15 +02:00
commit 9bbe253810
32 changed files with 2182 additions and 0 deletions

51
python/fces_native.cpp Normal file
View File

@@ -0,0 +1,51 @@
/**
* @file fces_native.cpp
* @brief Python bindings for FCES-native via pybind11.
*
* Exposes FCESOptimizer as a drop-in replacement for the Python implementation.
*
* Usage:
* from fces_native import FCESOptimizer
* opt = FCESOptimizer(model.parameters(), lr=1.6e-3, population_size=200)
*/
#include <pybind11/pybind11.h>
#include <pybind11/stl.h>
#include <torch/extension.h>
#include "fces/optimizer.hpp"
#include "fces/config.hpp"
namespace py = pybind11;
PYBIND11_MODULE(fces_native, m) {
m.doc() = "FCES-native: High-performance C++ FCES optimizer";
py::class_<fces::FCESConfig>(m, "FCESConfig")
.def(py::init<>())
.def_readwrite("lr", &fces::FCESConfig::lr)
.def_readwrite("population_size", &fces::FCESConfig::population_size)
.def_readwrite("total_steps", &fces::FCESConfig::total_steps)
.def_readwrite("grokking_coefficient", &fces::FCESConfig::grokking_coefficient)
.def_readwrite("direct_construction", &fces::FCESConfig::direct_construction);
py::class_<fces::FCESOptimizer>(m, "FCESOptimizer")
.def(py::init<std::vector<torch::Tensor>, fces::FCESConfig>(),
py::arg("params"),
py::arg("config") = fces::FCESConfig{})
.def("step", &fces::FCESOptimizer::step)
.def("update_fitness", &fces::FCESOptimizer::update_fitness)
.def("backup_to_ram", &fces::FCESOptimizer::backup_to_ram)
.def("restore_from_ram", &fces::FCESOptimizer::restore_from_ram)
.def("step_count", &fces::FCESOptimizer::step_count)
.def("calculate_sparsity", &fces::FCESOptimizer::calculate_sparsity)
.def("zero_grad", [](fces::FCESOptimizer& self) {
for (auto& group : self.param_groups()) {
for (auto& p : group.params()) {
if (p.grad().defined()) {
p.grad().zero_();
}
}
}
});
}

27
python/setup.py Normal file
View File

@@ -0,0 +1,27 @@
from setuptools import setup
from torch.utils.cpp_extension import BuildExtension, CppExtension
setup(
name="fces_native",
version="0.1.0",
description="High-performance C++ FCES optimizer (Python bindings)",
ext_modules=[
CppExtension(
name="fces_native",
sources=[
"fces_native.cpp",
"../src/config.cpp",
"../src/controller.cpp",
"../src/population.cpp",
"../src/fitness.cpp",
"../src/evolution.cpp",
"../src/spectral.cpp",
"../src/oscillation.cpp",
"../src/optimizer.cpp",
"../src/telemetry.cpp",
],
include_dirs=["../include"],
),
],
cmdclass={"build_ext": BuildExtension},
)