[Mosaic] apply_vector_layout C++: Fix check in vector.broadcast rule

PiperOrigin-RevId: 580115242
jax-ml · Nov 7, 2023 · c5d6df4 · c5d6df4
1 parent 5a1731c
commit c5d6df4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jaxlib/mosaic/dialect/tpu/transforms/apply_vector_layout.cc b/jaxlib/mosaic/dialect/tpu/transforms/apply_vector_layout.cc
@@ -1888,7 +1888,7 @@ LogicalResult vector_broadcast_rule(RewriteContext &ctx, Operation &op,
         return op.emitOpError("Not implemented: unsupported tiling");
       }
       int64_t num_tiles = layout_in.tilesPerVreg(ctx.target_shape);
-      CHECK(!(*(dim_eq.end() - 1) || !(*(dim_eq.end() - 2))));
+      CHECK(!*(dim_eq.end() - 1) || !*(dim_eq.end() - 2));
       if (*(dim_eq.end() - 1)) {  // Sublane broadcast
         if (num_tiles != 1) {
           return op.emitOpError(