def testDoNotCompileScalarConstGraph(self): with self.session() as sess: def my_graph(a, b): with ops.device("/device:IPU:0"): x = math_ops.add(a, b) return x with ops.device('cpu'): a = 2 b = 3 out = ipu.ipu_compiler.compile(my_graph, [a, b]) report = ReportJSON(self, sess) report.reset() result = sess.run(out) report.parse_log() report.assert_contains_no_compile_event() self.assertEqual(result, [5])
def testDoNotCompileScalarElementWiseGraphWithParameter(self): with self.session() as sess: def my_graph(a, b): with ops.device("/device:IPU:0"): x = math_ops.add(a, b) return x with ops.device('cpu'): a = array_ops.placeholder(np.int32, name="a") b = array_ops.placeholder(np.int32, name="b") out = ipu.ipu_compiler.compile(my_graph, [a, b]) report = ReportJSON(self, sess) report.reset() fd = {a: np.int32(2), b: np.int32(3)} result = sess.run(out, fd) report.parse_log() report.assert_contains_no_compile_event() self.assertAllClose(result, [5])