Searched refs:ary_out (Results 1 – 2 of 2) sorted by relevance
| /dports/devel/py-numba/numba-0.51.2/numba/cuda/tests/cudapy/ |
| H A D | test_warp_ops.py | 45 def use_vote_sync_all(ary_in, ary_out): argument
48 ary_out[i] = pred
51 def use_vote_sync_any(ary_in, ary_out): argument
54 ary_out[i] = pred
57 def use_vote_sync_eq(ary_in, ary_out): argument
60 ary_out[i] = pred
69 def use_match_any_sync(ary_in, ary_out): argument
72 ary_out[i] = ballot
75 def use_match_all_sync(ary_in, ary_out): argument
78 ary_out[i] = ballot if pred else 0
[all …]
|
| H A D | test_sync.py | 58 def use_syncthreads_count(ary_in, ary_out): argument
63 def use_syncthreads_and(ary_in, ary_out): argument
65 ary_out[i] = cuda.syncthreads_and(ary_in[i])
68 def use_syncthreads_or(ary_in, ary_out): argument
70 ary_out[i] = cuda.syncthreads_or(ary_in[i])
141 compiled[1, 72](ary_in, ary_out)
149 compiled[1, nelem](ary_in, ary_out)
150 self.assertTrue(np.all(ary_out == 1))
152 compiled[1, nelem](ary_in, ary_out)
160 compiled[1, nelem](ary_in, ary_out)
[all …]
|