Skip to content

Commit

Permalink
revert async mr test
Browse files Browse the repository at this point in the history
  • Loading branch information
rongou committed Jun 17, 2024
1 parent 085d31f commit 6349f29
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions tests/mr/device/cuda_async_mr_tests.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ using cuda_async_mr = rmm::mr::cuda_async_memory_resource;
static_assert(cuda::mr::resource_with<cuda_async_mr, cuda::mr::device_accessible>);
static_assert(cuda::mr::async_resource_with<cuda_async_mr, cuda::mr::device_accessible>);

class SystemMRTest : public ::testing::Test {
class AsyncMRTest : public ::testing::Test {
protected:
void SetUp() override
{
Expand All @@ -38,7 +38,7 @@ class SystemMRTest : public ::testing::Test {
}
};

TEST_F(SystemMRTest, ThrowIfNotSupported)
TEST_F(AsyncMRTest, ThrowIfNotSupported)
{
auto construct_mr = []() { cuda_async_mr mr; };
#ifndef RMM_CUDA_MALLOC_ASYNC_SUPPORT
Expand All @@ -49,7 +49,7 @@ TEST_F(SystemMRTest, ThrowIfNotSupported)
}

#if defined(RMM_CUDA_MALLOC_ASYNC_SUPPORT)
TEST_F(SystemMRTest, ExplicitInitialPoolSize)
TEST_F(AsyncMRTest, ExplicitInitialPoolSize)
{
const auto pool_init_size{100};
cuda_async_mr mr{pool_init_size};
Expand All @@ -58,7 +58,7 @@ TEST_F(SystemMRTest, ExplicitInitialPoolSize)
RMM_CUDA_TRY(cudaDeviceSynchronize());
}

TEST_F(SystemMRTest, ExplicitReleaseThreshold)
TEST_F(AsyncMRTest, ExplicitReleaseThreshold)
{
const auto pool_init_size{100};
const auto pool_release_threshold{1000};
Expand All @@ -68,7 +68,7 @@ TEST_F(SystemMRTest, ExplicitReleaseThreshold)
RMM_CUDA_TRY(cudaDeviceSynchronize());
}

TEST_F(SystemMRTest, DifferentPoolsUnequal)
TEST_F(AsyncMRTest, DifferentPoolsUnequal)
{
const auto pool_init_size{100};
const auto pool_release_threshold{1000};
Expand Down

0 comments on commit 6349f29

Please sign in to comment.