Searched refs:gpuStreamSynchronize (Results 1 – 8 of 8) sorted by relevance
58 assert(gpuStreamSynchronize(gpu_device.stream()) == gpuSuccess); in test_gpu_nullary()104 assert(gpuStreamSynchronize(gpu_device.stream()) == gpuSuccess); in test_gpu_elementwise_small()156 assert(gpuStreamSynchronize(gpu_device.stream()) == gpuSuccess); in test_gpu_elementwise()199 assert(gpuStreamSynchronize(gpu_device.stream()) == gpuSuccess); in test_gpu_props()238 assert(gpuStreamSynchronize(gpu_device.stream()) == gpuSuccess); in test_gpu_reduction()353 assert(gpuStreamSynchronize(gpu_device.stream()) == gpuSuccess); in test_gpu_convolution_1d()406 assert(gpuStreamSynchronize(gpu_device.stream()) == gpuSuccess); in test_gpu_convolution_inner_dim_col_major_1d()459 assert(gpuStreamSynchronize(gpu_device.stream()) == gpuSuccess); in test_gpu_convolution_inner_dim_row_major_1d()513 assert(gpuStreamSynchronize(gpu_device.stream()) == gpuSuccess); in test_gpu_convolution_2d()577 assert(gpuStreamSynchronize(gpu_device.stream()) == gpuSuccess); in test_gpu_convolution_3d()[all …]
57 assert(gpuStreamSynchronize(gpu_device.stream()) == gpuSuccess); in test_gpu_simple_argmax()116 assert(gpuStreamSynchronize(gpu_device.stream()) == gpuSuccess); in test_gpu_argmax_dim()144 assert(gpuStreamSynchronize(gpu_device.stream()) == gpuSuccess); in test_gpu_argmax_dim()205 assert(gpuStreamSynchronize(gpu_device.stream()) == gpuSuccess); in test_gpu_argmin_dim()233 assert(gpuStreamSynchronize(gpu_device.stream()) == gpuSuccess); in test_gpu_argmin_dim()
39 assert(gpuStreamSynchronize(gpu_device.stream()) == gpuSuccess); in test_gpu_random_uniform()65 assert(gpuStreamSynchronize(gpu_device.stream()) == gpuSuccess); in test_gpu_random_normal()
344 assert(gpuStreamSynchronize(context.device().stream()) == gpuSuccess); in test_gpu()355 assert(gpuStreamSynchronize(context.device().stream()) == gpuSuccess); in test_gpu()373 assert(gpuStreamSynchronize(context.device().stream()) == gpuSuccess); in test_gpu()
31 #define gpuStreamSynchronize hipStreamSynchronize macro37 #define gpuStreamSynchronize cudaStreamSynchronize macro73 gpuStreamSynchronize(stream); in XlaPythonGpuCallback()128 gpuStreamSynchronize(stream); in XlaPythonGpuCallback()
51 #define gpuStreamSynchronize hipStreamSynchronize macro81 #define gpuStreamSynchronize cudaStreamSynchronize macro
36 #undef gpuStreamSynchronize
306 gpuError_t err = gpuStreamSynchronize(stream_->stream()); in synchronize()