jax.experimental.pallas.mosaic_gpu.try_cluster_cancel# jax.experimental.pallas.mosaic_gpu.try_cluster_cancel(result_ref, barrier)[源代码]# 发起一个异步请求，从网格中领取一个新的工作单元。 它允许 SM 通过原子地取消网格中待处理集群的启动并声明其 CTA ID 作为下一个工作单元，从而动态获取工作。 参数: result_ref (_Ref) – 一个 16 字节结果将被存储的 SMEM ref。 barrier (_Ref) – 用于协调查询完成的屏障。 返回类型: 无 另请参阅 jax.experimental.pallas.mosaic_gpu.query_cluster_cancel()
