@@ -562,13 +562,13 @@ def wrap_in_typecast(self, actual_type, needed_dtype, s):
562562
563563 return super ().wrap_in_typecast (actual_type , needed_dtype , s )
564564
565- def map_group_hw_index (self , expr , type_context : str ):
565+ def map_group_hw_index (self , expr , type_context ):
566566 return var ("gid" )(expr .axis )
567567
568- def map_local_hw_index (self , expr , type_context : str ):
568+ def map_local_hw_index (self , expr , type_context ):
569569 return var ("lid" )(expr .axis )
570570
571- def map_variable (self , expr , type_context : str ):
571+ def map_variable (self , expr , type_context ):
572572
573573 if self .codegen_state .vectorization_info :
574574 if self .codegen_state .vectorization_info .iname == expr .name :
@@ -584,7 +584,7 @@ def map_variable(self, expr, type_context: str):
584584 return Literal (vector_literal )
585585 return super ().map_variable (expr , type_context )
586586
587- def map_if (self , expr , type_context : str ):
587+ def map_if (self , expr , type_context ):
588588 from loopy .types import to_loopy_type
589589 result_type = self .infer_type (expr )
590590 conditional_needed_loopy_type = to_loopy_type (np .bool_ )
0 commit comments