mirror of
https://github.com/Kitware/CMake.git
synced 2026-01-05 13:20:47 -06:00
d7204e649e
Run the `clang-format.bash` script to update all our C and C++ code to a new style defined by `.clang-format`. Use `clang-format` version 6.0. * If you reached this commit for a line in `git blame`, re-run the blame operation starting at the parent of this commit to see older history for the content. * See the parent commit for instructions to rebase a change across this style transition commit.
62 lines
1.4 KiB
Plaintext
62 lines
1.4 KiB
Plaintext
|
|
#include <iostream>
|
|
|
|
#include "file1.h"
|
|
#include "file2.h"
|
|
|
|
#ifdef _WIN32
|
|
# define EXPORT __declspec(dllexport)
|
|
# define IMPORT __declspec(dllimport)
|
|
#else
|
|
# define EXPORT
|
|
# define IMPORT
|
|
#endif
|
|
|
|
result_type __device__ file1_func(int x);
|
|
result_type_dynamic __device__ file2_func(int x);
|
|
|
|
IMPORT void __host__ cuda_dynamic_lib_func();
|
|
|
|
static __global__ void mixed_kernel(result_type* r, int x)
|
|
{
|
|
*r = file1_func(x);
|
|
result_type_dynamic rd = file2_func(x);
|
|
}
|
|
|
|
EXPORT int mixed_launch_kernel(int x)
|
|
{
|
|
cuda_dynamic_lib_func();
|
|
|
|
result_type* r;
|
|
cudaError_t err = cudaMallocManaged(&r, sizeof(result_type));
|
|
if (err != cudaSuccess) {
|
|
std::cerr << "mixed_launch_kernel: cudaMallocManaged failed: "
|
|
<< cudaGetErrorString(err) << std::endl;
|
|
return x;
|
|
}
|
|
|
|
mixed_kernel<<<1, 1>>>(r, x);
|
|
err = cudaGetLastError();
|
|
if (err != cudaSuccess) {
|
|
std::cerr << "mixed_kernel [SYNC] failed: " << cudaGetErrorString(err)
|
|
<< std::endl;
|
|
return x;
|
|
}
|
|
err = cudaDeviceSynchronize();
|
|
if (err != cudaSuccess) {
|
|
std::cerr << "mixed_kernel [ASYNC] failed: "
|
|
<< cudaGetErrorString(cudaGetLastError()) << std::endl;
|
|
return x;
|
|
}
|
|
|
|
int result = r->sum;
|
|
err = cudaFree(r);
|
|
if (err != cudaSuccess) {
|
|
std::cerr << "mixed_launch_kernel: cudaFree failed: "
|
|
<< cudaGetErrorString(err) << std::endl;
|
|
return x;
|
|
}
|
|
|
|
return result;
|
|
}
|