@@ -514,18 +514,15 @@ def _get_csr_matmul_prg(self, out_ndim: int) -> loopy.TranslationUnit:
514514 "iel_lbound" ,
515515 shape = (),
516516 address_space = lp .AddressSpace .GLOBAL ,
517- # FIXME: Need to do anything with tags?
518517 ),
519518 "iel_ubound" : lp .TemporaryVariable (
520519 "iel_ubound" ,
521520 shape = (),
522521 address_space = lp .AddressSpace .GLOBAL ,
523- # FIXME: Need to do anything with tags?
524522 )}
525523
526524 from loopy .kernel .instruction import make_assignment
527525 from pymbolic import var
528- # FIXME: Need tags for any of these?
529526 instructions : list [lp .Assignment | lp .CallInstruction ] = [
530527 make_assignment (
531528 (var ("iel_lbound" ),),
@@ -590,7 +587,6 @@ def _get_csr_matmul_prg(self, out_ndim: int) -> loopy.TranslationUnit:
590587 options = _DEFAULT_LOOPY_OPTIONS ,
591588 default_order = lp .auto ,
592589 default_offset = lp .auto ,
593- # FIXME: Need to do anything with tags?
594590 )
595591
596592 idx_dtype = knl .default_entrypoint .index_dtype
@@ -619,8 +615,6 @@ def _matmul(ary: ArrayOrScalar) -> ArrayOrScalar:
619615 prg , elem_values = x1 .elem_values ,
620616 elem_col_indices = x1 .elem_col_indices ,
621617 row_starts = x1 .row_starts , array = ary )["out" ]
622- # FIXME
623- # return self.tag(tagged, out_ary)
624618 return out_ary
625619
626620 return cast ("ArrayOrContainer" , rec_map_container (_matmul , x2 ))
0 commit comments