[SOT] Support string slice (#70405)

SigureMo · web-flow · commit 517f854b9f87 · 2024-12-24T10:08:32.000+08:00
diff --git a/python/paddle/jit/sot/opcode_translator/executor/variables/basic.py b/python/paddle/jit/sot/opcode_translator/executor/variables/basic.py
@@ -163,6 +163,18 @@ def bool_not(self):
             not bool(self.get_py_value()), self.graph, DummyTracker([self])
         )
 
+    def getitem(self, key):
+        track_vars: list[VariableBase] = [self]
+        if self.get_py_type() is not str:
+            raise InnerError(
+                f"getitem can only be applied to a str variable, but got {self.get_py_type()}"
+            )
+        if isinstance(key, VariableBase):
+            track_vars.append(key)
+            key = key.get_py_value()
+        retval = self.value[key]
+        return ConstantVariable(retval, self.graph, DummyTracker(track_vars))
+
     def str(self):
         return ConstantVariable(
             str(self.value), self.graph, DummyTracker([self])
diff --git a/test/sot/test_15_slice.py b/test/sot/test_15_slice.py
@@ -136,5 +136,16 @@ def test_layer_list_slice(self):
         self.assert_results(layer_list_slice, net, x)
 
 
+@check_no_breakgraph
+def string_slice(x: str):
+    return x[2:7:2] + x[1:5] + x[4]
+
+
+class TestStringSlice(TestCaseBase):
+    def test_string_slice(self):
+        x = "1234567"
+        self.assert_results(string_slice, x)
+
+
 if __name__ == "__main__":
     unittest.main()
