Numba'fied函数在for循环中的计算结果与在纯Python中的计算结果不同

Question

我正在使用numba来加快功能，包括许多for循环。 比它们在外部被称为循环。

在循环中调用时，其中一个函数的求值方法有所不同，因此不会引起误解。

功能：

def collison(u, v, f, feq, omega, w, cx, cy, n, m):
    for i in range(0, n):
        for j in range(0, n):
            t1 = u[i,j]*u[i,j] + v[i,j]*v[i,j]
            for k in range(9):
                t2 = u[i,j]*cx[k] + v[i,j]*cy[k]
                feq[k,i,j] = rho[i,j]*w[k]*(1. + 3.*t2 + 4.5*t2**2 - 1.5*t1)
                f[k,i,j] = omega*feq[k, i, j] + (1. - omega)*f[k, i, j]
    return f

u ， v -ndarray nxm f ， feq -ndarray 9 xnxm

我只是在声明之前添加装饰器： @nb.jit(nopython=True)

循环：

for tstep in range(mstep):
    fk = collison(u, v, fw, feq, omega, w, cx, cy, n, m)
    fs = stream(fk, n, m)
    fw = wbrzeg(fs, n, m, u0)
    rho, u, v = rhouv(fw, rho, u, v, cx, cy, n, m)

可能是在调用collision时未更新fw吗？

Answer 1

在固定函数中，Numba在编译函数时将rho视为常数。 这与始终考虑更新的rho的非附加功能不同。

见该条目在Numba常见问题。

Answer 2

我不确定您要在这里做什么，但是在tstep循环中，应该通过冲突来更改fw，但是直到wbrzeg再次更改它后，它才真正被使用（无论如何）。 因此，在您的代码中，如果fw被冲突更新实际上并不重要...

在您的示例中，这可能只是一种打字错误，但修复可能会帮助实际上了解您的问题的人（即不是我）...