numba 相关问题

Numba是一个开源的NumPy感知的Python优化编译器。

在 numba 中使用并行化对于一种算法更好,但对于另一种类似算法则不然?

我正在模拟具有相同质量和随机速度的 N 个粒子。我想计算系统的总能量和总动量。我考虑了 4 种方法并想进行基准测试...

回答 1 投票 0

inspect_asm 没有输出

我有这个简单的MWE: 从 numba 导入 njit @njit def 添加(a,b): 返回 a + b # 现在让我们检查“add()”函数的汇编代码。 对于 add.inspect_asm().items() 中的 k、v: 优先...

回答 2 投票 0

优化从 Astropy 的地心真黄道天空坐标中提取纬度/经度

我面临着一个挑战,这个挑战应该很简单,但事实证明 Astropy 库相当复杂。具体来说,我需要经常计算 SkyC 的经度和纬度...

回答 1 投票 0

如何并行化矩阵上的简单循环?

我有一个大矩阵,我想输出矩阵中元素小于0的所有索引。我在numba中有这个MWE: 将 numba 导入为 nb 将 numpy 导入为 np A = np.random.random(si...

回答 1 投票 0

使用更少的内存在两个矩阵之间执行异或运算的最有效方法

我想通过节省内存使用来在两个2D numpy.ndarray之间执行异或运算。对于每一行 u_values,我想对每一行 v_values 执行异或运算,依此类推。对于e...

回答 1 投票 0

Dockerfile 因 numba 安装的 llvm-config 错误而失败

我使用 pypy 库的 Dockerfile 失败,并出现 FileNotFoundError: [Errno 2] No such file or directory: 'llvm-config' 安装 llvmlite 时,numba 的依赖项列在我的 requests.t...

回答 9 投票 0

当 numba jitclass 包含 jitted 函数时,如何指定它的字段?

我想创建一个 numba jitclass,其属性包含任何 jitted 函数。 # 在另一个文件中定义的简单 jitted 函数 @njit def my_function(x): x = x + 1 返回x ...

回答 1 投票 0

记录类型为字符串的numba jitclass

v3 变量是字符串值。我无法使用下面的代码运行,这会出现错误。 将 numpy 导入为 np 将 pandas 导入为 pd 从 numba.experimental 导入 jitclass 从 numba 导入类型 导入操作系统 ...

回答 1 投票 0

将 jit 与 numpy linspace 函数一起使用时出错

我正在尝试使用 jit 来编译我的代码。目前,除了 Numba 之外我唯一使用的库是 Numpy。当指定 nopython = True 时,它会抛出一堆 linspace 数组错误。我有

回答 1 投票 0

传递类变量是否会停止 Numba 中的并行化?

我有一个包装方法来调用 Numba 兼容函数。在下面的代码中,方法 get_neighbours_wrapper() 只是调用 Numba 函数 get_neighbours_Numba() 的包装器。 我想...

回答 1 投票 0

如何有效地加快 Pandas 中我心爱代码的循环速度?

我一直在考虑使用多处理、cyton 和 numba 来加速我的代码的以下部分,但我就是无法弄清楚。我目前的结论是我的代码是无效的,尽管它

回答 1 投票 0

使用 numba 进行 PI 的蒙特卡洛模拟对于最少的点数来说是最慢的?

作为家庭作业练习的一部分,我正在用 Python 实现 pi 的蒙特卡罗模拟。我正在使用 Numba 来加速和并行化计算。从我之前进行的测试中,我发现...

回答 1 投票 0

numba 按类型调度

我想在 numba 函数中分派第二个参数的类型,但失败了。 如果它是一个整数,那么应该返回一个向量, 如果它本身是一个整数数组,...

回答 1 投票 0

为什么这个带有列表输入的简单 numba 函数这么慢

进口numba 从输入导入列表 @numba.njit def test(a: List[int]) -> int: 返回1 测试([i 为 i 在范围(2_000_000)]) 需要 2 秒并随着列表的大小线性缩放。 瓦拉皮...

回答 1 投票 0

如何并行化/加速令人尴尬的并行 numba 代码?

我有以下代码: @nb.njit(缓存=True) def find_two_largest(arr): # 初始化第一大和第二大元素 如果 arr[0] >= arr[1]: 最大= arr[0]

回答 1 投票 0

将数组参数发送到 numba 函数

我有以下代码,它将数组作为参数发送到 numba 函数: 将 numpy 导入为 np 从 numba 导入 njit,float64 A = [(0.0182286178413157,-1.2904019395416308), ( 0.

回答 1 投票 0

如何使用numba加速多维Longsumexp和softmax

我想使用多项式logit计算softmax/概率,同时使用longsumexp以避免溢出。使用 numba 带来 2-3 倍的加速。我在这里可以做得更好吗?另外,当我使用 fastmath=True 时,我...

回答 1 投票 0

将 2d 字符串 numpy 数组转换为 3d int 数组的快速方法

我有一个非常大的 numpy 数组,其中包含以下条目: [['0/1''2/0'] ['3/0''1/4']] 我想转换它/获取一个带有 3d 数组的数组,例如 [[[0 1] [2 0]] [[3 0] [1 4]]] 数组很宽,所以...

回答 2 投票 0

如何使用Numba CUDA JIT装饰器?

我已按照本教程使用 Numba CUDA JIT 装饰器:https://www.youtube.com/watch?v=-lcWV4wkHsk&t=510s。 这是我的Python代码: 将 numpy 导入为 np from timeit import default_timer as t...

回答 1 投票 0


© www.soinside.com 2019 - 2024. All rights reserved.