[SOT] Support for simulation of enum types

[0x3878f]

PR Category

Execute Infrastructure

PR Types

New features

Description

Support for simulation of enum types:

• Add variable EnumVariable
• operator.eq and operator.ne for EnumVariables

pcard-67164

[paddle-bot]

你的PR提交成功，感谢你对开源项目的贡献!
请关注后续CI自动化测试结果，详情请参考Paddle-CI手册。
Your PR has been submitted. Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.

[SigureMo]

TODO：之后有时间将 (VariableBase, VariableBase) ne 转发到 eq 取反，这样就不必每个都单独写一遍了

这里可以把 TODO 记到代码里

[0x3878f]

好的

[SigureMo]

这里的 self.get_py_value().__class__.__name__ 会在哪里用到呢？f"{self.value!s}" 会用到是么？ 看来是的：

>>> from enum import Enum >>> >>> class Foo(Enum): ... A = 1 ... B = 2 ... >>> str(Foo.A) 'Foo.A' >>> repr(Foo.A) '<Foo.A: 1>' >>> 

这里其实是有一点风险的，毕竟依赖于 str 的实现，而且还有多个 Foo 冲突的风险，比如

# a.py class Foo(Enum): A: 1 def fn(foo: Foo): ... # b.py class Foo(Enum): B: 2 def fn(foo: Foo): ...

这里 a.fn.__globals__['Foo'] 和 b.fn.__globals__['Foo'] 完全不同，但是最终混合到生成的 guard 里会统一变成 Foo，后者会把前者覆盖，那在访问前者 Foo.A 时就会挂掉了

[SigureMo]

这里建议维护一个 name -> enum class 的映射

known_enum_classes = {} def get_enum_class_id(enum_class: type[Enum]): class_name = enum_class.__name__ known_enum_classes.setdefault(class_name, []) same_name_enums = known_enum_classes[class_name] id = 0 for i, cls in enumerate(same_name_enums): if enum_class == cls: id = i break else: id = len(same_name_enums) same_name_enums.append(enum_class) return id enum_class = instance.__class__ class_name = enum_class.__name__ enum_class_id = get_enum_class_id(enum_class) extern_var_name = f"__{class_name}_{enum_class_id}" # tracker 用 f"{extern_var_name}.{value.name}"

[0x3878f]

没有考虑到类名重复的问题，已修改

[SigureMo]

LGTMeow

[SigureMo]

LGTMeow