Skip to content

Commit

Permalink
Change map_direct fusion test to map_selection
Browse files Browse the repository at this point in the history
  • Loading branch information
tomwhite committed Nov 7, 2024
1 parent 7c097ad commit 7c09447
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions cubed/tests/test_optimization.py
Original file line number Diff line number Diff line change
Expand Up @@ -116,9 +116,9 @@ def test_fusion_transpose(spec, opt_fn):
)


def test_fusion_map_direct(spec):
# test that operations after a map_direct operation (indexing) can be fused
# with the map_direct operation
def test_fusion_map_selection(spec):
# test that operations after a map_selection operation (indexing) can be fused
# with the map_selection operation
# this is only true for the (default) multiple_inputs_optimize_dag optimize function
a = xp.asarray([[1, 2, 3], [4, 5, 6], [7, 8, 9]], chunks=(2, 2), spec=spec)
b = a[1:, :]
Expand Down

0 comments on commit 7c09447

Please sign in to comment.