Skip to content

Commit a3f3486

Browse files
committed
remove FIXMEs about tags
1 parent 0ca16ab commit a3f3486

1 file changed

Lines changed: 0 additions & 6 deletions

File tree

arraycontext/context.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)