physarum / local_bind.cpp

add Physarum flow solver (shortest path / network); card leads with it

0a16898 verified 8 days ago

659 Bytes

	// JIT binding for load_local.py; the published build registers via torch_binding.cpp.
	#include <torch/extension.h>

	void physarum_step(torch::Tensor trail, torch::Tensor tmp,
	torch::Tensor ax, torch::Tensor ay, torch::Tensor ah,
	int64_t seed, double sense_dist, double sense_ang,
	double turn, double speed, double deposit, double decay);

	void flow_cg(torch::Tensor cE, torch::Tensor cS, torch::Tensor b,
	torch::Tensor p, int64_t ground, int64_t max_iters, double tol);

	PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
	m.def("physarum_step", &physarum_step);
	m.def("flow_cg", &flow_cg);
	}