Searched refs:complex_bytes (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_complex_cwise_ops_cuda.cu | 25 std::size_t complex_bytes = kNumItems * sizeof(std::complex<T>); in test_cuda_complex_cwise_ops() local
30 cudaMalloc((void**)(&d_in1), complex_bytes); in test_cuda_complex_cwise_ops()
31 cudaMalloc((void**)(&d_in2), complex_bytes); in test_cuda_complex_cwise_ops()
32 cudaMalloc((void**)(&d_out), complex_bytes); in test_cuda_complex_cwise_ops()
78 assert(cudaMemcpyAsync(actual.data(), d_out, complex_bytes, cudaMemcpyDeviceToHost, in test_cuda_complex_cwise_ops()
|
| D | cxx11_tensor_complex_cuda.cu | 29 std::size_t complex_bytes = in1.size() * sizeof(std::complex<float>); in test_cuda_nullary() local
34 cudaMalloc((void**)(&d_in1), complex_bytes); in test_cuda_nullary()
35 cudaMalloc((void**)(&d_in2), complex_bytes); in test_cuda_nullary()
37 cudaMemcpy(d_in1, in1.data(), complex_bytes, cudaMemcpyHostToDevice); in test_cuda_nullary()
38 cudaMemcpy(d_in2, in2.data(), complex_bytes, cudaMemcpyHostToDevice); in test_cuda_nullary()
56 assert(cudaMemcpyAsync(new1.data(), d_in1, complex_bytes, cudaMemcpyDeviceToHost, in test_cuda_nullary()
|