手写 kernel 计算 Sigma 初值
一开始这个循环实在是太耗时了，也分配了许多显存。这个更改能将这一块的代码提速500倍，应该也能减少显存分配。
/cc @heroxbd 看看这个是否能解决 #12 (closed) ？我觉得 #12 (closed) 之所以还没有到极限就无法分配显存也和这一块有关。
一开始这个循环实在是太耗时了，也分配了许多显存。这个更改能将这一块的代码提速500倍，应该也能减少显存分配。
/cc @heroxbd 看看这个是否能解决 #12 (closed) ？我觉得 #12 (closed) 之所以还没有到极限就无法分配显存也和这一块有关。