浏览代码

Add PrimitiveFinished to JIT globals

jonathanvdc 8 年之前
父节点
当前提交
a5619a718c
共有 1 个文件被更改,包括 4 次插入2 次删除
  1. 4 2
      kernel/modelverse_jit/jit.py

+ 4 - 2
kernel/modelverse_jit/jit.py

@@ -15,7 +15,9 @@ class ModelverseJit(object):
         self.todo_entry_points = set()
         self.no_jit_entry_points = set()
         self.jitted_entry_points = {}
-        self.jit_globals = {}
+        self.jit_globals = {
+            'PrimitiveFinished' : primitive_functions.PrimitiveFinished
+        }
         self.jit_count = 0
 
     def mark_entry_point(self, body_id):
@@ -82,7 +84,7 @@ class ModelverseJit(object):
         compiled_function = self.jit_globals[constructed_function.name]
 
         print(constructed_function)
-        
+
         # Save the compiled function so we can reuse it later.
         self.jitted_entry_points[body_id] = compiled_function
         raise primitive_functions.PrimitiveFinished(compiled_function)