def run(self, name):
self.compile_graph()
profiler = get_profiler()
profiler.start()
reset_jit()
i = self.code_mapping[name]
retval = self.interp.eval_graph(self.graph, [i])
return retval
def run(self, name):
self.compile_graph()
profiler = get_profiler()
profiler.start()
reset_jit()
i = self.code_mapping[name]
retval = self.interp.eval_graph(self.graph, [i])
return retval
def test_reduce_axis_compile_only_once(self):
self.compile_graph()
reset_jit()
i = self.code_mapping['axissum']
# run it twice
retval = self.interp.eval_graph(self.graph, [i])
retval = self.interp.eval_graph(self.graph, [i])
# check that we got only one loop
assert len(get_stats().loops) == 1
def test_reduce_compile_only_once(self):
self.compile_graph()
reset_jit()
i = self.code_mapping['reduce']
# run it twice
retval = self.interp.eval_graph(self.graph, [i])
assert retval == sum(range(1, 11))
retval = self.interp.eval_graph(self.graph, [i])
assert retval == sum(range(1, 11))
# check that we got only one loop
assert len(get_stats().loops) == 1
self.check_vectorized(2, 0)
def test_reduce_compile_only_once(self):
self.compile_graph()
reset_jit()
i = self.code_mapping['reduce']
# run it twice
retval = self.interp.eval_graph(self.graph, [i])
assert retval == sum(range(1,11))
retval = self.interp.eval_graph(self.graph, [i])
assert retval == sum(range(1,11))
# check that we got only one loop
assert len(get_stats().loops) == 1
self.check_vectorized(2, 0)
