an.tsouchlos/ba-thesis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

Added more Proximal Decoder template specializations

Browse Source
This commit is contained in:
Andreas Tsouchlos 2022-11-28 19:22:23 +01:00
parent 7d0e4abfae
commit f76728a3d9
2 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
sw/cpp/src/proximal.h
View File

@ -1,7 +1,5 @@
#pragma once #pragma once
#define EIGEN_STACK_ALLOCATION_LIMIT 524288
#include <Eigen/Dense> #include <Eigen/Dense>
#include <bit> #include <bit>
#include <iostream> #include <iostream>

12
sw/cpp/src/python_interface.cpp
View File

@ -1,3 +1,5 @@
#define EIGEN_STACK_ALLOCATION_LIMIT 1048576
#include "proximal.h" #include "proximal.h"
#include <pybind11/pybind11.h> #include <pybind11/pybind11.h>
@ -28,11 +30,17 @@ using namespace pybind11::literals;
PYBIND11_MODULE(cpp_decoders, proximal) { PYBIND11_MODULE(cpp_decoders, proximal) {
proximal.doc() = "Proximal decoder"; proximal.doc() = "Proximal decoder";
DEF_PROXIMAL_DECODER("ProximalDecoder_7_4", 4, 7) DEF_PROXIMAL_DECODER("ProximalDecoder_7_3", 3, 7)
DEF_PROXIMAL_DECODER("ProximalDecoder_31_20", 20, 31)
DEF_PROXIMAL_DECODER("ProximalDecoder_31_6", 6, 31)
DEF_PROXIMAL_DECODER("ProximalDecoder_96_48", 48, 96) DEF_PROXIMAL_DECODER("ProximalDecoder_96_48", 48, 96)
DEF_PROXIMAL_DECODER("ProximalDecoder_204_102", 102, 204) DEF_PROXIMAL_DECODER("ProximalDecoder_204_102", 102, 204)
DEF_PROXIMAL_DECODER("ProximalDecoder_408_204", 204, 408)
DEF_PROXIMAL_DECODER("ProximalDecoder_Dynamic", Eigen::Dynamic, DEF_PROXIMAL_DECODER("ProximalDecoder_Dynamic", Eigen::Dynamic,
Eigen::Dynamic) Eigen::Dynamic)
py::register_exception<std::runtime_error>(proximal, "CppException"); py::register_exception<std::runtime_error>(
proximal, "CppException: std::runtime_error");
py::register_exception<std::bad_alloc>(proximal,
"CppException: std::bad_alloc");
} }