コード例 #1
0
def shuffle_down(val, width):
    tid = roc.get_local_id(0)
    roc.wavebarrier()
    idx = (tid + width) % WAVESIZE
    res = roc.ds_permute(idx, val)
    return res
コード例 #2
0
ファイル: test_intrinsics.py プロジェクト: esc/numba 
def shuffle_down(val, width):
    tid = roc.get_local_id(0)
    roc.wavebarrier()
    idx = (tid - width) % _WAVESIZE
    res = roc.ds_permute(idx, val)
    return res
コード例 #3
0
 def foo(inp, mask, out):
     tid = roc.get_local_id(0)
     out[tid] = roc.ds_permute(inp[tid], mask[tid])
コード例 #4
0
ファイル: test_scan.py プロジェクト: esc/numba 
 def foo(inp, mask, out):
     tid = roc.get_local_id(0)
     out[tid] = roc.ds_permute(inp[tid], mask[tid])