Skip to content

Commit

Permalink
Add test
Browse files Browse the repository at this point in the history
  • Loading branch information
Daniel Arndt authored and masterleinad committed Aug 28, 2023
1 parent 64a9b3d commit e627b2e
Show file tree
Hide file tree
Showing 2 changed files with 40 additions and 0 deletions.
6 changes: 6 additions & 0 deletions core/unit_test/CMakeLists.txt
Expand Up @@ -763,6 +763,12 @@ if(Kokkos_ENABLE_CUDA)
UnitTestMain.cpp
cuda/TestCuda_InterOp_Streams.cpp
)
KOKKOS_ADD_EXECUTABLE_AND_TEST(
CoreUnitTest_CudaInterOpStreamsMultiGPU
SOURCES
UnitTestMain.cpp
cuda/TestCuda_InterOp_StreamsMultiGPU.cpp
)
KOKKOS_ADD_EXECUTABLE_AND_TEST(
CoreUnitTest_CudaGraph
SOURCES
Expand Down
34 changes: 34 additions & 0 deletions core/unit_test/cuda/TestCuda_InterOp_StreamsMultiGPU.cpp
@@ -0,0 +1,34 @@
//@HEADER
// ************************************************************************
//
// Kokkos v. 4.0
// Copyright (2022) National Technology & Engineering
// Solutions of Sandia, LLC (NTESS).
//
// Under the terms of Contract DE-NA0003525 with NTESS,
// the U.S. Government retains certain rights in this software.
//
// Part of Kokkos, under the Apache License v2.0 with LLVM Exceptions.
// See https://kokkos.org/LICENSE for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//@HEADER

#include <Kokkos_Core.hpp>
#include <TestCuda_Category.hpp>
#include <gtest/gtest.h>

namespace Test {
// Test Interoperability with Cuda Streams and muliple GPUs.
TEST(cuda, raw_cuda_streams) {
Kokkos::ScopeGuard scope_guard;

cudaStream_t stream;
KOKKOS_IMPL_CUDA_SAFE_CALL(cudaStreamCreate(&stream));
{
TEST_EXECSPACE cuda_instance(TEST_EXECSPACE().cuda_device(), stream);
ASSERT_EQ(cuda_instance.cuda_device(), TEST_EXECSPACE().cuda_device());
}
KOKKOS_IMPL_CUDA_SAFE_CALL(cudaStreamDestroy(stream));
}
} // namespace Test

0 comments on commit e627b2e

Please sign in to comment.