Skip to content

Commit c03a070

Browse files
committed
edit _Py_uop_sym_get_probable_type and update test
1 parent cc75b3d commit c03a070

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

Python/optimizer_symbols.c

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
#ifdef _Py_TIER2
1+
ifdef _Py_TIER2
22

33
#include "Python.h"
44

@@ -829,7 +829,7 @@ _Py_uop_sym_get_probable_type(JitOptRef ref)
829829
case JIT_SYM_KNOWN_VALUE_TAG:
830830
return _Py_uop_sym_get_type(ref);
831831
case JIT_SYM_RECORDED_GEN_FUNC_TAG:
832-
return NULL;
832+
return &PyGen_Type;
833833
case JIT_SYM_RECORDED_VALUE_TAG:
834834
return Py_TYPE(sym->recorded_value.value);
835835
case JIT_SYM_RECORDED_TYPE_TAG:
@@ -2211,7 +2211,7 @@ _Py_uop_symbols_test(PyObject *Py_UNUSED(self), PyObject *Py_UNUSED(ignored))
22112211
JitOptRef rg1 = _Py_uop_sym_new_unknown(ctx);
22122212
_Py_uop_sym_set_recorded_gen_func(ctx, rg1, func);
22132213
TEST_PREDICATE(!_Py_uop_sym_matches_type(rg1, &PyGen_Type), "recorded gen func treated as generator");
2214-
TEST_PREDICATE(_Py_uop_sym_get_probable_type(rg1) == NULL, "recorded gen func treated as generator");
2214+
TEST_PREDICATE(_Py_uop_sym_get_probable_type(rg1) == &PyGen_Type, "recorded gen func not treated as generator");
22152215
TEST_PREDICATE(_Py_uop_sym_get_const(ctx, rg1) == NULL, "recorded gen func is treated as known value");
22162216

22172217
/* Test that setting type narrows correctly */

0 commit comments

Comments
 (0)