From 0d9bdc101cdc724a125d3557cf5cf4684bed14cf Mon Sep 17 00:00:00 2001
From: Sebastien Bourdeauducq <sb@m-labs.hk>
Date: Tue, 17 Jun 2014 18:52:48 +0200
Subject: [PATCH] inline: check inlined functions are on the same core device

---
 artiq/compiler/inline.py | 11 ++++++-----
 artiq/devices/core.py    |  2 +-
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/artiq/compiler/inline.py b/artiq/compiler/inline.py
index 516a0447e..81b37ba72 100644
--- a/artiq/compiler/inline.py
+++ b/artiq/compiler/inline.py
@@ -107,7 +107,8 @@ _embeddable_calls = {
 }
 
 class _ReferenceReplacer(ast.NodeTransformer):
-	def __init__(self, rm, obj, funcname):
+	def __init__(self, core, rm, obj, funcname):
+		self.core = core
 		self.rm = rm
 		self.obj = obj
 		self.funcname = funcname
@@ -131,9 +132,9 @@ class _ReferenceReplacer(ast.NodeTransformer):
 			new_func = ast.Name(func.__name__, ast.Load())
 			return ast.Call(func=new_func, args=new_args,
 				keywords=[], starargs=None, kwargs=None)
-		elif hasattr(func, "k_function_info"):
+		elif hasattr(func, "k_function_info") and getattr(func.__self__, func.k_function_info.core_name) is self.core:
 			args = [func.__self__] + new_args
-			inlined, _ = inline(func.k_function_info.k_function, args, dict(), self.rm)
+			inlined, _ = inline(self.core, func.k_function_info.k_function, args, dict(), self.rm)
 			return inlined
 		else:
 			args = [ast.Str("rpc"), ast.Num(self.rm.rpc_map[func])]
@@ -192,7 +193,7 @@ def _initialize_function_params(funcdef, k_args, k_kwargs, rm):
 			param_init.append(ast.Assign(targets=[target], value=value))
 	return param_init
 
-def inline(k_function, k_args, k_kwargs, rm=None):
+def inline(core, k_function, k_args, k_kwargs, rm=None):
 	if rm is None:
 		rm = _ReferenceManager()
 
@@ -202,7 +203,7 @@ def inline(k_function, k_args, k_kwargs, rm=None):
 
 	obj = k_args[0]
 	funcname = funcdef.name
-	rr = _ReferenceReplacer(rm, obj, funcname)
+	rr = _ReferenceReplacer(core, rm, obj, funcname)
 	rr.visit(funcdef)
 
 	funcdef.body[0:0] = param_init
diff --git a/artiq/devices/core.py b/artiq/devices/core.py
index 8c7fd30eb..66eb365ab 100644
--- a/artiq/devices/core.py
+++ b/artiq/devices/core.py
@@ -5,7 +5,7 @@ from artiq.compiler.unparse import Unparser
 
 class Core:
 	def run(self, k_function, k_args, k_kwargs):
-		stmts, rpc_map = inline(k_function, k_args, k_kwargs)
+		stmts, rpc_map = inline(self, k_function, k_args, k_kwargs)
 
 		print("=========================")
 		print(" Inlined")