Searched refs:dev_a (Results 1 – 4 of 4) sorted by relevance
| /third_party/boost/libs/fiber/examples/cuda/ |
| D | single_stream.cu | 45 int * dev_a, * dev_b, * dev_c; in main() local
46 cudaMalloc( & dev_a, size * sizeof( int) ); in main()
56 … cudaMemcpyAsync( dev_a, host_a + i, size * sizeof( int), cudaMemcpyHostToDevice, stream); in main()
58 vector_add<<< size / 256, 256, 0, stream >>>( dev_a, dev_b, dev_c, size); in main()
68 cudaFree( dev_a); in main()
|
| /third_party/boost/libs/fiber/examples/hip/ |
| D | single_stream.cpp | 45 int * dev_a, * dev_b, * dev_c; in main() local
46 hipMalloc( & dev_a, size * sizeof( int) ); in main()
56 … hipMemcpyAsync( dev_a, host_a + i, size * sizeof( int), hipMemcpyHostToDevice, stream); in main()
58 … hipLaunchKernel( vector_add, dim3(size / 256), dim3(256), 0, stream, dev_a, dev_b, dev_c, size); in main()
68 hipFree( dev_a); in main()
|
| /third_party/boost/libs/fiber/doc/ |
| D | cuda.qbk | 46 int * dev_a, * dev_b, * dev_c;
47 cudaMalloc( & dev_a, size * sizeof( int) );
57 … cudaMemcpyAsync( dev_a, host_a + i, size * sizeof( int), cudaMemcpyHostToDevice, stream);
59 kernel<<< size / 256, 256, 0, stream >>>( size, dev_a, dev_b, dev_c);
69 cudaFree( dev_a);
|
| D | hip.qbk | 47 int * dev_a, * dev_b, * dev_c;
48 hipMalloc( & dev_a, size * sizeof( int) );
58 … hipMemcpyAsync( dev_a, host_a + i, size * sizeof( int), hipMemcpyHostToDevice, stream);
60 … hipLaunchKernel(kernel, dim3(size / 256), dim3(256), 0, stream, size, dev_a, dev_b, dev_c);
70 hipFree( dev_a);
|