numba 中的逻辑右移

Question

以下代码

@jit(uint32(uint32))
def foo(x): 
    return (x >> 6)

print(hex(foo(0xffffffff)))

输出

0xffffffff

因为 numba 使用算术右移，尽管操作数被定义为无符号。

有什么方法可以传达它以使用逻辑移位？

实际上，从下面 JoshAdel 的回答中，我学到了一种解决方法 - 使用内部 64 位变量，其逻辑和算术移位不会有所不同。像这样：

@jit(uint32(uint32), locals = {'t': uint64})
def foo1(x):
    t = x 
    return t << 8

@jit(uint32(uint32), locals = {'t': uint64})
def foo2(x):
    t = x 
    return t >> 8

print(hex(foo1(0xffffffff)))
print(hex(foo2(0xffffffff)))

打印出来的

0xffffff00
0xffffff

但一个体面的编译器应该明白，>> 应用于 unsigned 数字会导致逻辑移位。

Answer 1

我从使用 Numba 0.27 的纯 python 和 numba jitted 版本中得到了相同的答案：

import numba as nb

def foo(x): 
    return (x >> 6)

foo_nb = nb.jit()(foo)

print(hex(foo(0xffffffff)))
print(hex(foo_nb(0xffffffff)))

结果：

0x3ffffff
0x3ffffff

以下也产生相同的结果（以确保我们不会退回到对象模式并实际使用完整的 jit）：

foo_nb2 = nb.jit('uint(uint)', nopython=True)(foo)
print(hex(foo_nb2(0xffffffff)))

# prints 0x3ffffff

您使用的是哪个版本的 Numba？