Skip to content

lucyddm: 减小 rocm 下的 batch size

王宇逸 requested to merge fix/lucyddm-batchsize into master

Closes #40 (closed)

根据 rocFFT 打开 plan log(指定 ROCFFT_LAYER 为8即可,实际上我设定了13=1|4|8,打开了 trace, profile, plan)可以看出来 grid params 之中直接粗暴地将 grid x 或者 grid z 设置成了 batch size,导致可能超过 rocm 允许的上限。这一问题没有在 N 卡上发现,因此修复方案只针对 rocm。

/cc @xuchuang @wuyy

Merge request reports