forked from M-Labs/artiq
26 lines
857 B
Python
26 lines
857 B
Python
from artiq.compiler.inline import inline
|
|
from artiq.compiler.fold_constants import fold_constants
|
|
from artiq.compiler.unroll_loops import unroll_loops
|
|
from artiq.compiler.interleave import interleave
|
|
from artiq.compiler.lower_time import lower_time
|
|
from artiq.compiler.ir import get_runtime_binary
|
|
|
|
class Core:
|
|
def __init__(self, core_com, runtime_env=None):
|
|
if runtime_env is None:
|
|
runtime_env = core_com.get_runtime_env()
|
|
self.runtime_env = runtime_env
|
|
self.core_com = core_com
|
|
|
|
def run(self, k_function, k_args, k_kwargs):
|
|
stmts, rpc_map = inline(self, k_function, k_args, k_kwargs)
|
|
fold_constants(stmts)
|
|
unroll_loops(stmts, 50)
|
|
interleave(stmts)
|
|
lower_time(stmts, self.runtime_env.ref_period)
|
|
fold_constants(stmts)
|
|
|
|
binary = get_runtime_binary(self.runtime_env, stmts)
|
|
self.core_com.run(binary)
|
|
self.core_com.serve(rpc_map)
|