@@ -185,9 +185,9 @@ class Lane:
185
185
# The wave this lane belongs to.
186
186
_wave : Wave
187
187
# The callstack of this lane. Each tuple represents 1 call.
188
- # The first element is the IP the function will return to.
189
- # The second element is the callback to call to store the return value
190
- # into the correct register.
188
+ # The first element is the IP the function will return to.
189
+ # The second element is the callback to call to store the return value
190
+ # into the correct register.
191
191
_callstack : list [Tuple [InstructionPointer , Callable [[Any ], None ]]]
192
192
193
193
_previous_bb : Optional [BasicBlock ]
@@ -234,7 +234,7 @@ def set_register(self, name: str, value: Any) -> None:
234
234
self ._registers [name ] = value
235
235
236
236
# Get the value in register "name" in this lane.
237
- # if allow_undef is true, fetching an unknown register won't fail.
237
+ # If allow_undef is true, fetching an unknown register won't fail.
238
238
def get_register (self , name : str , allow_undef : bool = False ) -> Optional [Any ]:
239
239
if allow_undef and name not in self ._registers :
240
240
return None
@@ -404,7 +404,7 @@ class Wave:
404
404
# The instructions scheduled for execution.
405
405
_tasks : Task
406
406
# The actual requirements to comply with when executing instructions.
407
- # e .g: the set of lanes required to merge before executing the merge block.
407
+ # E .g: the set of lanes required to merge before executing the merge block.
408
408
_convergence_requirements : list [ConvergenceRequirement ]
409
409
# The indices of the active lanes for the current executing instruction.
410
410
_active_lane_indices : set [int ]
@@ -502,7 +502,7 @@ def _has_tasks(self) -> bool:
502
502
def get_first_active_lane_index (self ) -> int :
503
503
return min (self ._active_lane_indices )
504
504
505
- # Broadcast the given value to all active lane registers' .
505
+ # Broadcast the given value to all active lane registers.
506
506
def broadcast_register (self , register : str , value : Any ) -> None :
507
507
for tid in self ._active_lane_indices :
508
508
self ._lanes [tid ].set_register (register , value )
0 commit comments