summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--Tools/cases_generator/test_generator.py175
1 files changed, 47 insertions, 128 deletions
diff --git a/Tools/cases_generator/test_generator.py b/Tools/cases_generator/test_generator.py
index 412d3d1..f8c7997 100644
--- a/Tools/cases_generator/test_generator.py
+++ b/Tools/cases_generator/test_generator.py
@@ -43,7 +43,8 @@ def run_cases_test(input: str, expected: str):
temp_input.write(generate_cases.END_MARKER)
temp_input.flush()
temp_output = tempfile.NamedTemporaryFile("w+")
- a = generate_cases.Analyzer(temp_input.name, temp_output.name)
+ temp_metadata = tempfile.NamedTemporaryFile("w+")
+ a = generate_cases.Analyzer([temp_input.name], temp_output.name, temp_metadata.name)
a.parse()
a.analyze()
if a.errors:
@@ -84,7 +85,7 @@ def test_inst_one_pop():
"""
output = """
TARGET(OP) {
- PyObject *value = PEEK(1);
+ PyObject *value = stack_pointer[-1];
spam();
STACK_SHRINK(1);
DISPATCH();
@@ -103,7 +104,7 @@ def test_inst_one_push():
PyObject *res;
spam();
STACK_GROW(1);
- POKE(1, res);
+ stack_pointer[-1] = res;
DISPATCH();
}
"""
@@ -117,10 +118,10 @@ def test_inst_one_push_one_pop():
"""
output = """
TARGET(OP) {
- PyObject *value = PEEK(1);
+ PyObject *value = stack_pointer[-1];
PyObject *res;
spam();
- POKE(1, res);
+ stack_pointer[-1] = res;
DISPATCH();
}
"""
@@ -134,12 +135,12 @@ def test_binary_op():
"""
output = """
TARGET(OP) {
- PyObject *right = PEEK(1);
- PyObject *left = PEEK(2);
+ PyObject *right = stack_pointer[-1];
+ PyObject *left = stack_pointer[-2];
PyObject *res;
spam();
STACK_SHRINK(1);
- POKE(1, res);
+ stack_pointer[-1] = res;
DISPATCH();
}
"""
@@ -153,11 +154,11 @@ def test_overlap():
"""
output = """
TARGET(OP) {
- PyObject *right = PEEK(1);
- PyObject *left = PEEK(2);
+ PyObject *right = stack_pointer[-1];
+ PyObject *left = stack_pointer[-2];
PyObject *result;
spam();
- POKE(1, result);
+ stack_pointer[-1] = result;
DISPATCH();
}
"""
@@ -167,11 +168,8 @@ def test_predictions_and_eval_breaker():
input = """
inst(OP1, (--)) {
}
- inst(OP2, (--)) {
- }
inst(OP3, (arg -- res)) {
DEOPT_IF(xxx, OP1);
- PREDICT(OP2);
CHECK_EVAL_BREAKER();
}
"""
@@ -181,17 +179,11 @@ def test_predictions_and_eval_breaker():
DISPATCH();
}
- TARGET(OP2) {
- PREDICTED(OP2);
- DISPATCH();
- }
-
TARGET(OP3) {
- PyObject *arg = PEEK(1);
+ PyObject *arg = stack_pointer[-1];
PyObject *res;
DEOPT_IF(xxx, OP1);
- POKE(1, res);
- PREDICT(OP2);
+ stack_pointer[-1] = res;
CHECK_EVAL_BREAKER();
DISPATCH();
}
@@ -234,12 +226,12 @@ def test_error_if_pop():
"""
output = """
TARGET(OP) {
- PyObject *right = PEEK(1);
- PyObject *left = PEEK(2);
+ PyObject *right = stack_pointer[-1];
+ PyObject *left = stack_pointer[-2];
PyObject *res;
if (cond) goto pop_2_label;
STACK_SHRINK(1);
- POKE(1, res);
+ stack_pointer[-1] = res;
DISPATCH();
}
"""
@@ -252,11 +244,11 @@ def test_cache_effect():
"""
output = """
TARGET(OP) {
- PyObject *value = PEEK(1);
+ PyObject *value = stack_pointer[-1];
uint16_t counter = read_u16(&next_instr[0].cache);
uint32_t extra = read_u32(&next_instr[1].cache);
STACK_SHRINK(1);
- JUMPBY(3);
+ next_instr += 3;
DISPATCH();
}
"""
@@ -275,59 +267,6 @@ def test_suppress_dispatch():
"""
run_cases_test(input, output)
-def test_super_instruction():
- # TODO: Test cache effect
- input = """
- inst(OP1, (counter/1, arg --)) {
- op1();
- }
- inst(OP2, (extra/2, arg --)) {
- op2();
- }
- super(OP) = OP1 + OP2;
- """
- output = """
- TARGET(OP1) {
- PyObject *arg = PEEK(1);
- uint16_t counter = read_u16(&next_instr[0].cache);
- op1();
- STACK_SHRINK(1);
- JUMPBY(1);
- DISPATCH();
- }
-
- TARGET(OP2) {
- PyObject *arg = PEEK(1);
- uint32_t extra = read_u32(&next_instr[0].cache);
- op2();
- STACK_SHRINK(1);
- JUMPBY(2);
- DISPATCH();
- }
-
- TARGET(OP) {
- PyObject *_tmp_1 = PEEK(1);
- PyObject *_tmp_2 = PEEK(2);
- {
- PyObject *arg = _tmp_1;
- uint16_t counter = read_u16(&next_instr[0].cache);
- op1();
- }
- JUMPBY(1);
- NEXTOPARG();
- JUMPBY(1);
- {
- PyObject *arg = _tmp_2;
- uint32_t extra = read_u32(&next_instr[0].cache);
- op2();
- }
- JUMPBY(2);
- STACK_SHRINK(2);
- DISPATCH();
- }
- """
- run_cases_test(input, output)
-
def test_macro_instruction():
input = """
inst(OP1, (counter/1, left, right -- left, right)) {
@@ -344,18 +283,18 @@ def test_macro_instruction():
"""
output = """
TARGET(OP1) {
- PyObject *right = PEEK(1);
- PyObject *left = PEEK(2);
+ PyObject *right = stack_pointer[-1];
+ PyObject *left = stack_pointer[-2];
uint16_t counter = read_u16(&next_instr[0].cache);
op1(left, right);
- JUMPBY(1);
+ next_instr += 1;
DISPATCH();
}
TARGET(OP) {
- PyObject *_tmp_1 = PEEK(1);
- PyObject *_tmp_2 = PEEK(2);
- PyObject *_tmp_3 = PEEK(3);
+ PyObject *_tmp_1 = stack_pointer[-1];
+ PyObject *_tmp_2 = stack_pointer[-2];
+ PyObject *_tmp_3 = stack_pointer[-3];
{
PyObject *right = _tmp_1;
PyObject *left = _tmp_2;
@@ -373,22 +312,22 @@ def test_macro_instruction():
res = op2(arg2, left, right);
_tmp_3 = res;
}
- JUMPBY(5);
+ next_instr += 5;
static_assert(INLINE_CACHE_ENTRIES_OP == 5, "incorrect cache size");
STACK_SHRINK(2);
- POKE(1, _tmp_3);
+ stack_pointer[-1] = _tmp_3;
DISPATCH();
}
TARGET(OP3) {
- PyObject *right = PEEK(1);
- PyObject *left = PEEK(2);
- PyObject *arg2 = PEEK(3);
+ PyObject *right = stack_pointer[-1];
+ PyObject *left = stack_pointer[-2];
+ PyObject *arg2 = stack_pointer[-3];
PyObject *res;
res = op3(arg2, left, right);
STACK_SHRINK(2);
- POKE(1, res);
- JUMPBY(5);
+ stack_pointer[-1] = res;
+ next_instr += 5;
DISPATCH();
}
"""
@@ -402,9 +341,9 @@ def test_array_input():
"""
output = """
TARGET(OP) {
- PyObject *above = PEEK(1);
- PyObject **values = &PEEK(1 + oparg*2);
- PyObject *below = PEEK(2 + oparg*2);
+ PyObject *above = stack_pointer[-1];
+ PyObject **values = (stack_pointer - (1 + oparg*2));
+ PyObject *below = stack_pointer[-(2 + oparg*2)];
spam();
STACK_SHRINK(oparg*2);
STACK_SHRINK(2);
@@ -426,8 +365,8 @@ def test_array_output():
PyObject *above;
spam(values, oparg);
STACK_GROW(oparg*3);
- POKE(1, above);
- POKE(2 + oparg*3, below);
+ stack_pointer[-1] = above;
+ stack_pointer[-(2 + oparg*3)] = below;
DISPATCH();
}
"""
@@ -441,11 +380,11 @@ def test_array_input_output():
"""
output = """
TARGET(OP) {
- PyObject **values = &PEEK(oparg);
+ PyObject **values = (stack_pointer - oparg);
PyObject *above;
spam(values, oparg);
STACK_GROW(1);
- POKE(1, above);
+ stack_pointer[-1] = above;
DISPATCH();
}
"""
@@ -459,8 +398,8 @@ def test_array_error_if():
"""
output = """
TARGET(OP) {
- PyObject **values = &PEEK(oparg);
- PyObject *extra = PEEK(1 + oparg);
+ PyObject **values = (stack_pointer - oparg);
+ PyObject *extra = stack_pointer[-(1 + oparg)];
if (oparg == 0) { STACK_SHRINK(oparg); goto pop_1_somewhere; }
STACK_SHRINK(oparg);
STACK_SHRINK(1);
@@ -469,26 +408,6 @@ def test_array_error_if():
"""
run_cases_test(input, output)
-def test_register():
- input = """
- register inst(OP, (counter/1, left, right -- result)) {
- result = op(left, right);
- }
- """
- output = """
- TARGET(OP) {
- PyObject *left = REG(oparg1);
- PyObject *right = REG(oparg2);
- PyObject *result;
- uint16_t counter = read_u16(&next_instr[0].cache);
- result = op(left, right);
- Py_XSETREF(REG(oparg3), result);
- JUMPBY(1);
- DISPATCH();
- }
- """
- run_cases_test(input, output)
-
def test_cond_effect():
input = """
inst(OP, (aa, input if ((oparg & 1) == 1), cc -- xx, output if (oparg & 2), zz)) {
@@ -497,18 +416,18 @@ def test_cond_effect():
"""
output = """
TARGET(OP) {
- PyObject *cc = PEEK(1);
- PyObject *input = ((oparg & 1) == 1) ? PEEK(1 + (((oparg & 1) == 1) ? 1 : 0)) : NULL;
- PyObject *aa = PEEK(2 + (((oparg & 1) == 1) ? 1 : 0));
+ PyObject *cc = stack_pointer[-1];
+ PyObject *input = ((oparg & 1) == 1) ? stack_pointer[-(1 + (((oparg & 1) == 1) ? 1 : 0))] : NULL;
+ PyObject *aa = stack_pointer[-(2 + (((oparg & 1) == 1) ? 1 : 0))];
PyObject *xx;
PyObject *output = NULL;
PyObject *zz;
output = spam(oparg, input);
STACK_SHRINK((((oparg & 1) == 1) ? 1 : 0));
STACK_GROW(((oparg & 2) ? 1 : 0));
- POKE(1, zz);
- if (oparg & 2) { POKE(1 + ((oparg & 2) ? 1 : 0), output); }
- POKE(2 + ((oparg & 2) ? 1 : 0), xx);
+ stack_pointer[-1] = zz;
+ if (oparg & 2) { stack_pointer[-(1 + ((oparg & 2) ? 1 : 0))] = output; }
+ stack_pointer[-(2 + ((oparg & 2) ? 1 : 0))] = xx;
DISPATCH();
}
"""