logique de décalage à droite dans numba

Question

de code suivant

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

print(hex(foo(0xffffffff))) 

sorties

0xffffffff 

parce numba utilise décalage droit arithmétique malgré le fait que l'opérande est définie comme étant non signé.

Y a-t-il un moyen de le transmettre pour utiliser le décalage logique? En fait, à partir de la réponse de JoshAdel ci-dessous, j'ai appris une solution de contournement - en utilisant une variable interne de 64 bits pour laquelle les décalages logiques et arithmétiques ne différeront pas. Comme ceci:

@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))) 

qui imprime

0xffffff00 
0xffffff 

Mais un compilateur décent doit comprendre que >> appliqué à un nombre unsigned entraînera un changement logique.

Answer 1

je reçois la même réponse du python pur et Numba versions jitted utilisant Numba 0.27:

import numba as nb 

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

foo_nb = nb.jit()(foo) 

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

résultats dans:

0x3ffffff 
0x3ffffff 

Ce qui suit produit également le même résultat (pour nous assurer que nous n'êtes pas retomber en mode objet et en utilisant effectivement la pleine JIT):

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

# prints 0x3ffffff 

W la version chapeau de Numba utilisez-vous?