Torch-Spyre博士的感应器决定如何在Spyre的内核之间分配每个内核的工作，并管理片上便签簿和片外存储器之间的张量驻留，后端编译器将这些决定转化为硬件可以实际运行的东西。它们之间的契约使用KTIR(内核瓦片IR)表示，这是一种建立在加速器数据并行视图上的MLIR表示，其中每个计算瓦片可以看到每一个内存，单个逻辑张量可以实时分布在几个内核的便签簿上。…
从Ktir到SpyreCode：IBM Spyre的MLIR数据流后端
From KTIR to SpyreCode: An MLIR Dataflow Backend for IBM Spyre
专题配图图片仅供信息参考图片仅供参考版权归原作者摘要
Torch-Spyre博士的感应器决定如何在Spyre的内核之间分配每个内核的工作，并管理片上便签簿和片外存储器之间的张量驻留，后端编译器将这些决策转化为……
原始资料
PyTorch Blog · 中文编译 · 正文节选
查看来源原文 产品参数、测试条件及最新修订以来源原文为准。