jax.experimental.pallas.triton.debug_barrier#

jax.experimental.pallas.triton.debug_barrier()[源代码]#

同步网格中的所有内核执行。

返回类型: