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)