diff options
author | Adrian Kummerlaender | 2021-09-12 14:01:55 +0200 |
---|---|---|
committer | Adrian Kummerlaender | 2021-09-12 14:01:55 +0200 |
commit | 32dd41a728ce10113032e20955ba08f8de449857 (patch) | |
tree | 8a5f851a342590d05137b2c900a7d3c4e8c545b6 /tangle/magnus.cu | |
parent | 8bca21a550e0ef134d51c4c4c007720885d76791 (diff) | |
download | LiterateLB-32dd41a728ce10113032e20955ba08f8de449857.tar LiterateLB-32dd41a728ce10113032e20955ba08f8de449857.tar.gz LiterateLB-32dd41a728ce10113032e20955ba08f8de449857.tar.bz2 LiterateLB-32dd41a728ce10113032e20955ba08f8de449857.tar.lz LiterateLB-32dd41a728ce10113032e20955ba08f8de449857.tar.xz LiterateLB-32dd41a728ce10113032e20955ba08f8de449857.tar.zst LiterateLB-32dd41a728ce10113032e20955ba08f8de449857.zip |
Start using C++ cuda-api-wrapper instead of raw CUDA
Diffstat (limited to 'tangle/magnus.cu')
-rw-r--r-- | tangle/magnus.cu | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/tangle/magnus.cu b/tangle/magnus.cu index 08a4515..a98bcf3 100644 --- a/tangle/magnus.cu +++ b/tangle/magnus.cu @@ -13,7 +13,11 @@ using T = float; using DESCRIPTOR = descriptor::D2Q9; int main() { -cudaSetDevice(0); +if (cuda::device::count() == 0) { + std::cerr << "No CUDA devices on this system" << std::endl; + return -1; +} +auto current = cuda::device::current::get(); const descriptor::Cuboid<DESCRIPTOR> cuboid(1200, 500); Lattice<DESCRIPTOR,T> lattice(cuboid); @@ -64,7 +68,7 @@ auto inflow_mask = materials.mask_of_material(3); auto outflow_mask = materials.mask_of_material(4); auto edge_mask = materials.mask_of_material(5); -cudaDeviceSynchronize(); +cuda::synchronize(current); RenderWindow window("Magnus"); cudaSurfaceObject_t colormap; @@ -83,7 +87,7 @@ while (window.isOpen()) { lattice.apply<BouzidiO>(bouzidi.getCount(), bouzidi.getConfig()); lattice.stream(); if (iStep % 200 == 0) { - cudaDeviceSynchronize(); + cuda::synchronize(current); lattice.inspect<CollectMomentsF>(bulk_mask, moments_rho.device(), moments_u.device()); renderSliceViewToTexture<<< dim3(cuboid.nX / 32 + 1, cuboid.nY / 32 + 1), |