| 380 | raise ValueError("Only axis=None and axis=0 are supported for count") |
| 381 | |
| 382 | class CountPerBlockImpl: |
| 383 | def op(self): |
| 384 | return array_count |
| 385 | |
| 386 | def op2(self): |
| 387 | return _local_array_count |
| 388 | |
| 389 | def mpi_op(self): |
| 390 | from mpi4py import MPI |
| 391 | return MPI.SUM |
| 392 | |
| 393 | def default(self): |
| 394 | return numpy.float64(0) |
| 395 | |
| 396 | return _global_per_block(CountPerBlockImpl(), array, axis, controller) |
| 397 |