Skip to content

Commit c290317

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 bf6a643 commit c290317

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
@@ -596,6 +596,10 @@ class test_p2p_send_on_diff_device : public uct_p2p_test {
596596
protected:
597597
void init() override
598598
{
599+
if (!mem_buffer::is_mem_type_supported(UCS_MEMORY_TYPE_CUDA)) {
600+
UCS_TEST_SKIP_R("CUDA is not supported");
601+
}
602+
599603
ASSERT_EQ(cudaGetDeviceCount(&m_num_devices), cudaSuccess);
600604
if (m_num_devices < 2) {
601605
UCS_TEST_SKIP_R("less than two cuda devices available");

0 commit comments

Comments
 (0)