summaryrefslogtreecommitdiff
path: root/python
diff options
context:
space:
mode:
authorPeter LaFosse <peter@vector35.com>2021-09-16 10:33:22 -0400
committerPeter LaFosse <peter@vector35.com>2021-09-16 10:50:17 -0400
commita0a7b91801f087f2dc4e808b3061df4a2f6fc3fe (patch)
tree0541eda0dfd3573745bbfaecc00aa1e56d83911a /python
parent77f44cb3db3d8070fc9655508af22712325c538b (diff)
Fix type check error on self.regs/self.reg_stacks
Diffstat (limited to 'python')
-rw-r--r--python/architecture.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/python/architecture.py b/python/architecture.py
index 67814f6f..2dbbc5e1 100644
--- a/python/architecture.py
+++ b/python/architecture.py
@@ -316,13 +316,15 @@ class Architecture(metaclass=_ArchitectureMetaClass):
self._all_regs:Mapping[RegisterName, RegisterIndex] = {}
self._full_width_regs:Mapping[RegisterName, RegisterIndex] = {}
self._regs_by_index:Mapping[RegisterIndex, RegisterName] = {}
- self.regs:Mapping[RegisterName, RegisterInfo] = self.__class__.regs
+ self.regs = self.__class__.regs
+ assert self.regs is not None, "Custom Architecture doesn't specify a register map"
reg_index = RegisterIndex(0)
# Registers used for storage in register stacks must be sequential, so allocate these in order first
self._all_reg_stacks:Mapping[RegisterStackName, RegisterStackIndex] = {}
self._reg_stacks_by_index:Mapping[RegisterStackIndex, RegisterStackName] = {}
- self.reg_stacks:Mapping[RegisterStackName, RegisterStackInfo] = self.__class__.reg_stacks
+ self.reg_stacks = self.__class__.reg_stacks
+ assert self.regs is not None, "Custom Architecture doesn't specify a reg_stacks map"
reg_stack_index = RegisterStackIndex(0)
for reg_stack, info in self.reg_stacks.items():
for reg in info.storage_regs: