Aug 15, 2023 · Aug 2, 2023 · Aug 2, 2023 · Aug 2, 2023 · Aug 2, 2023 · Aug 2, 2023
diff --git a/Include/internal/pycore_opcode_metadata.h b/Include/internal/pycore_opcode_metadata.h
diff --git a/Include/internal/pycore_uops.h b/Include/internal/pycore_uops.h
 #  error "this header requires Py_BUILD_CORE define"
 #endif

 #define _Py_UOP_MAX_TRACE_LENGTH32
 #define _Py_UOP_MAX_TRACE_LENGTH256

 typedef struct {
    uint32_t opcode;
diff --git a/Lib/test/test_capi/test_misc.py b/Lib/test/test_capi/test_misc.py
            with self.assertRaises(StopIteration):
                next(it)


 class TestUopsOptimization(unittest.TestCase):

    def test_int_constant_propagation(self):
        def testfunc(loops):
            num = 0
            while num < loops:
                x = 0
                y = 1
                z = 2
                a = x + y + z + x + y + z + x + y + z
                num += 1
            return a

        opt = _testinternalcapi.get_uop_optimizer()
        res = None
        with temporary_optimizer(opt):
            res = testfunc(3)

        ex = get_first_executor(testfunc)
        self.assertIsNotNone(ex)
        self.assertEqual(res, 9)
        binop_count = [opname == "_BINARY_OP_ADD_INT" for opname, _, _ in ex]
        self.assertEqual(binop_count, 1)

 if __name__ == "__main__":
    unittest.main()
Original file line number	Diff line number	Diff line change
Expand Up		@@ -8,7 +8,7 @@ extern "C" {
		# error "this header requires Py_BUILD_CORE define"
		#endif

		#define _Py_UOP_MAX_TRACE_LENGTH32
		#define _Py_UOP_MAX_TRACE_LENGTH256
Fidget-Spinner marked this conversation as resolved. OutdatedShow resolvedHide resolved

		typedef struct {
		uint32_t opcode;
Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -2707,5 +2707,30 @@ def testfunc(it):
		with self.assertRaises(StopIteration):
		next(it)


		class TestUopsOptimization(unittest.TestCase):

		def test_int_constant_propagation(self):
		def testfunc(loops):
		num = 0
		while num < loops:
		x = 0
		y = 1
		z = 2
		a = x + y + z + x + y + z + x + y + z
		num += 1
		return a

		opt = _testinternalcapi.get_uop_optimizer()
		res = None
		with temporary_optimizer(opt):
		res = testfunc(3)

		ex = get_first_executor(testfunc)
		self.assertIsNotNone(ex)
		self.assertEqual(res, 9)
		binop_count = [opname == "_BINARY_OP_ADD_INT" for opname, _, _ in ex]
		self.assertEqual(binop_count, 1)

Fidget-Spinner marked this conversation as resolved. OutdatedShow resolvedHide resolved
		if __name__ == "__main__":
		unittest.main()