| D | cuda_common.h | 15 void run_on_cpu(const Kernel& ker, int n, const Input& in, Output& out) in run_on_cpu() argument
18 ker(i, in.data(), out.data()); in run_on_cpu()
24 void run_on_cuda_meta_kernel(const Kernel ker, int n, const Input* in, Output* out) in run_on_cuda_meta_kernel() argument
28 ker(i, in, out); in run_on_cuda_meta_kernel()
34 void run_on_cuda(const Kernel& ker, int n, const Input& in, Output& out) in run_on_cuda() argument
53 run_on_cuda_meta_kernel<<<Grids,Blocks>>>(ker, n, d_in, d_out); in run_on_cuda()
66 void run_and_compare_to_cuda(const Kernel& ker, int n, const Input& in, Output& out) in run_and_compare_to_cuda() argument
74 run_on_cpu (ker, n, in_ref, out_ref); in run_and_compare_to_cuda()
75 run_on_cuda(ker, n, in_cuda, out_cuda); in run_and_compare_to_cuda()
|