nicolas
/
copapy
mirror of https://github.com/Nonannet/copapy.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

stencil_db_from_package with arch_translation_table added

This commit is contained in:
Nicolas 2025-10-04 23:07:37 +02:00
parent 5d3702a1b2
commit b3eace3be4
2 changed files with 17 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
src/copapy/__init__.py
View File

@ -9,13 +9,21 @@ import platform
Operand = type['Net'] | float | int Operand = type['Net'] | float | int
def get_var_name(var: Any, scope: dict[str, Any] = globals()) -> list[str]: def get_var_name(var: Any, scope: dict[str, Any] = globals()) -> list[str]:
return [name for name, value in scope.items() if value is var] return [name for name, value in scope.items() if value is var]
_arch = platform.machine()
_stencil_data = pkgutil.get_data(__name__, f"obj/stencils_{_arch}_O3.o") def stencil_db_from_package(arch: str = 'native', optimization: str = 'O3') -> stencil_database:
assert _stencil_data arch_translation_table = {'ARM64': 'x86_64'}
generic_sdb = stencil_database(_stencil_data) if arch == 'native':
arch = arch_translation_table.get(platform.machine(), platform.machine())
stencil_data = pkgutil.get_data(__name__, f"obj/stencils_{arch}_{optimization}.o")
assert stencil_data, f"stencils_{arch}_{optimization} not found"
return stencil_database(stencil_data)
generic_sdb = stencil_db_from_package()
class Node: class Node:
@ -319,7 +327,7 @@ def get_nets(*inputs: Iterable[Iterable[Any]]) -> list[Net]:
def compile_to_instruction_list(node_list: Iterable[Node], sdb: stencil_database) -> tuple[binw.data_writer, dict[Net, tuple[int, int, str]]]: def compile_to_instruction_list(node_list: Iterable[Node], sdb: stencil_database) -> tuple[binw.data_writer, dict[Net, tuple[int, int, str]]]:
variables: dict[Net, tuple[int, int, str]] = dict() variables: dict[Net, tuple[int, int, str]] = dict()
ordered_ops = list(stable_toposort(get_all_dag_edges(node_list))) ordered_ops = list(stable_toposort(get_all_dag_edges(node_list)))
const_net_list = get_const_nets(ordered_ops) const_net_list = get_const_nets(ordered_ops)
output_ops = list(add_read_ops(ordered_ops)) output_ops = list(add_read_ops(ordered_ops))
@ -414,15 +422,10 @@ def compile_to_instruction_list(node_list: Iterable[Node], sdb: stencil_database
class Target(): class Target():
def __init__(self, arch: str = 'native', optimization: str = 'O3') -> None: def __init__(self, arch: str = 'native', optimization: str = 'O3') -> None:
if arch == 'native': self.sdb = stencil_db_from_package(arch, optimization)
arch = platform.machine()
stencil_data = pkgutil.get_data(__name__, f"obj/stencils_{arch}_{optimization}.o")
assert stencil_data
self.sdb = stencil_database(stencil_data)
self._variables: dict[Net, tuple[int, int, str]] = dict() self._variables: dict[Net, tuple[int, int, str]] = dict()
def compile(self, *variables: Net | list[Net]) -> None:
def compile(self, *variables: list[Net] | list[list[Net]]) -> None:
nodes: list[Node] = [] nodes: list[Node] = []
for s in variables: for s in variables:
if isinstance(s, Net): if isinstance(s, Net):
@ -431,13 +434,11 @@ class Target():
for net in s: for net in s:
assert isinstance(net, Net) assert isinstance(net, Net)
nodes.append(Write(net)) nodes.append(Write(net))
dw, self._variables = compile_to_instruction_list(nodes, self.sdb) dw, self._variables = compile_to_instruction_list(nodes, self.sdb)
dw.write_com(binw.Command.END_PROG) dw.write_com(binw.Command.END_PROG)
assert coparun(dw.get_data()) > 0 assert coparun(dw.get_data()) > 0
def run(self) -> None: def run(self) -> None:
# set entry point and run code # set entry point and run code
dw = binw.data_writer(self.sdb.byteorder) dw = binw.data_writer(self.sdb.byteorder)
@ -446,7 +447,6 @@ class Target():
dw.write_com(binw.Command.END_PROG) dw.write_com(binw.Command.END_PROG)
assert coparun(dw.get_data()) > 0 assert coparun(dw.get_data()) > 0
def read_value(self, net: Net) -> float | int: def read_value(self, net: Net) -> float | int:
assert net in self._variables, f"Variable {net} not found" assert net in self._variables, f"Variable {net} not found"
addr, lengths, var_type = self._variables[net] addr, lengths, var_type = self._variables[net]
@ -481,4 +481,4 @@ class Target():
dw.write_com(binw.Command.READ_DATA) dw.write_com(binw.Command.READ_DATA)
dw.write_int(addr) dw.write_int(addr)
dw.write_int(lengths) dw.write_int(lengths)
assert coparun(dw.get_data()) > 0 assert coparun(dw.get_data()) > 0

5
tests/test_coparun_module.py
View File

@ -1,7 +1,4 @@
from coparun_module import coparun from copapy import const, Target
from copapy import Write, const, Target
import copapy
from copapy import binwrite
def test_compile(): def test_compile():