Skip to content

Commit

Permalink
fix block_any, block_all
Browse files Browse the repository at this point in the history
  • Loading branch information
nikirchner committed Feb 3, 2025
1 parent 91d4cf4 commit b8d81a8
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions cyten/backends/fusion_tree_backend.py
Original file line number Diff line number Diff line change
Expand Up @@ -325,10 +325,10 @@ def diagonal_all(self, a: DiagonalTensor) -> bool:
# there are missing blocks. -> they contain False -> all(a) == False
return False
# now it is enough to check the existing blocks
return all(self.block_backend.all(b) for b in a.data.blocks)
return all(self.block_backend.block_all(b) for b in a.data.blocks)

def diagonal_any(self, a: DiagonalTensor) -> bool:
return any(self.block_backend.any(b) for b in a.data.blocks)
return any(self.block_backend.block_any(b) for b in a.data.blocks)

def diagonal_elementwise_binary(self, a: DiagonalTensor, b: DiagonalTensor, func,
func_kwargs, partial_zero_is_zero: bool) -> DiagonalData:
Expand Down

0 comments on commit b8d81a8

Please sign in to comment.