Skip to content

Commit 822ba27

Browse files
committed
trigger kernel test.
Signed-off-by: Qiliang Cui <[email protected]>
1 parent 8e187bc commit 822ba27

File tree

1 file changed

+1
-0
lines changed
  • tpu_inference/kernels/collectives

1 file changed

+1
-0
lines changed

tpu_inference/kernels/collectives/util.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
from jax.experimental.pallas import tpu as pltpu
88

99

10+
# Trigger a test
1011
def local_barrier(left_neighbor, right_neighbor, double_barrier=True):
1112
"""Performs a barrier with neighbors on the global barrier semaphore.
1213

0 commit comments

Comments
 (0)