小编Mat*_*lim的帖子

Numba 并行代码比顺序代码慢

我是 Numba 新手,我正在尝试使用 Numba (版本 0.54.1)在 Python 中实现旧的 Fortran 代码,但是当我添加时,parallel = True程序实际上变慢了。我的程序非常简单:我更改 L x L 网格中的位置 x 和 y,并对网格中的每个位置执行求和

import numpy as np
import numba as nb

@nb.njit(parallel=True)
def lyapunov_grid(x_grid, y_grid, k, N):
    L = len(x_grid)
    lypnv = np.zeros((L, L))
    for ii in nb.prange(L):
        for jj in range(L):
            x = x_grid[ii]
            y = y_grid[jj]
            beta0 = 0
            sumT11 = 0

            for j in range(N):
                y = (y - k*np.sin(x)) % (2*np.pi)
                x = (x + y) % (2*np.pi) …
Run Code Online (Sandbox Code Playgroud)

python parallel-processing performance numba

3
推荐指数
1
解决办法
1313
查看次数

标签 统计

numba ×1

parallel-processing ×1

performance ×1

python ×1