Implement POP_TOP__STORE_FAST super instruction

corona10 · May 28, 2023 · a96ed91 · a96ed91
1 parent 477a02c
commit a96ed91
Show file tree

Hide file tree

Showing 10 changed files with 506 additions and 471 deletions.
diff --git a/Include/internal/pycore_opcode.h b/Include/internal/pycore_opcode.h
diff --git a/Include/internal/pycore_opcode_utils.h b/Include/internal/pycore_opcode_utils.h
@@ -60,7 +60,8 @@ extern "C" {
          (opcode) == LOAD_FAST__LOAD_CONST || \
          (opcode) == LOAD_CONST__LOAD_FAST || \
          (opcode) == STORE_FAST__LOAD_FAST || \
-         (opcode) == STORE_FAST__STORE_FAST)
+         (opcode) == STORE_FAST__STORE_FAST) || \
+         (opcode) == POP_TOP__STORE_FAST
 
 
 #define LOG_BITS_PER_INT 5

diff --git a/Include/opcode.h b/Include/opcode.h
diff --git a/Lib/opcode.py b/Lib/opcode.py
@@ -435,6 +435,9 @@ def pseudo_op(name, op, real_ops):
     "SEND": [
         "SEND_GEN",
     ],
+    "POP_TOP": [
+        "POP_TOP__STORE_FAST"
+    ]
 }
 _specialized_instructions = [
     opcode for family in _specializations.values() for opcode in family

diff --git a/Python/bytecodes.c b/Python/bytecodes.c
@@ -214,6 +214,7 @@ dummy_func(
         super(STORE_FAST__LOAD_FAST)  = STORE_FAST + LOAD_FAST;
         super(STORE_FAST__STORE_FAST) = STORE_FAST + STORE_FAST;
         super(LOAD_CONST__LOAD_FAST) = LOAD_CONST + LOAD_FAST;
+        super(POP_TOP__STORE_FAST) = POP_TOP + STORE_FAST;
 
         inst(POP_TOP, (value --)) {
             DECREF_INPUTS();