lucyddm: 减小 rocm 下的 batch size
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。
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。