Set a bit within the predicate vector.
| 484 | |
| 485 | /// Set a bit within the predicate vector. |
| 486 | CUTLASS_HOST_DEVICE void set(int idx, bool value = true) { |
| 487 | int bit, word; |
| 488 | computeStorageOffset(word, bit, idx); |
| 489 | |
| 490 | Storage disable_mask = (~(Storage(1) << bit)); |
| 491 | Storage enable_mask = (Storage(value) << bit); |
| 492 | |
| 493 | storage(word) = ((storage(word) & disable_mask) | enable_mask); |
| 494 | } |
| 495 | |
| 496 | /// Computes the intersection of two identical predicate vectors. |
| 497 | CUTLASS_HOST_DEVICE PredicateVector &operator&=(PredicateVector const &predicates) { |