Searched refs:d_res_half (Results 1 – 1 of 1) sorted by relevance
31 bool* d_res_half = (bool*)gpu_device.allocate(num_elem * sizeof(bool)); in test_cuda_numext() local37 d_res_half, num_elem); in test_cuda_numext()47 gpu_device.memcpyDeviceToHost(half_prec.data(), d_res_half, num_elem*sizeof(bool)); in test_cuda_numext()57 gpu_device.deallocate(d_res_half); in test_cuda_numext()106 float* d_res_half = (float*)gpu_device.allocate(num_elem * sizeof(float)); in test_cuda_unary() local112 d_res_half, num_elem); in test_cuda_unary()122 gpu_device.memcpyDeviceToHost(half_prec.data(), d_res_half, num_elem*sizeof(float)); in test_cuda_unary()132 gpu_device.deallocate(d_res_half); in test_cuda_unary()144 float* d_res_half = (float*)gpu_device.allocate(num_elem * sizeof(float)); in test_cuda_elementwise() local152 d_res_half, num_elem); in test_cuda_elementwise()[all …]