Skip to content

Commit 652f7d0

Browse files
committed
TEST/CUDA: skip switch_cuda_device when CUDA unsupported
The patch created by Ilia Yastrebov Signed-off-by: Alexey Rivkin <[email protected]>
1 parent a2dd250 commit 652f7d0

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

test/gtest/uct/cuda/test_switch_cuda_device.cc

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -581,6 +581,10 @@ class test_p2p_send_on_diff_device : public uct_p2p_test {
581581
protected:
582582
void init() override
583583
{
584+
if (!mem_buffer::is_mem_type_supported(UCS_MEMORY_TYPE_CUDA)) {
585+
UCS_TEST_SKIP_R("CUDA is not supported");
586+
}
587+
584588
ASSERT_EQ(cudaGetDeviceCount(&m_num_devices), cudaSuccess);
585589
if (m_num_devices < 2) {
586590
UCS_TEST_SKIP_R("less than two cuda devices available");

0 commit comments

Comments
 (0)