Skip to content

Commit

Permalink
Skip gtests for new lanczos solver when CUDA version is 11.4 or below. (
Browse files Browse the repository at this point in the history
rapidsai#2520)

Medium- to longer term, we plan to fix this issue (ref:
rapidsai#2519).
  • Loading branch information
cjnolet authored Dec 6, 2024
1 parent 4ecd925 commit 0af5afb
Showing 1 changed file with 13 additions and 0 deletions.
13 changes: 13 additions & 0 deletions cpp/test/sparse/solver/lanczos.cu
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,7 @@
#include <raft/spectral/matrix_wrappers.hpp>
#include <raft/util/cudart_utils.hpp>

#include <cuda_runtime.h>
#include <driver_types.h>

#include <gtest/gtest.h>
Expand Down Expand Up @@ -265,6 +266,18 @@ class lanczos_tests : public ::testing::TestWithParam<lanczos_inputs<IndexType,

void Run()
{
int runtimeVersion;
cudaError_t result = cudaRuntimeGetVersion(&runtimeVersion);

if (result == cudaSuccess) {
int major = runtimeVersion / 1000;
int minor = (runtimeVersion % 1000) / 10;

// Skip gtests for CUDA 11.4.x and below because hard-coded results are causing issues.
// See https://github.com/rapidsai/raft/issues/2519 for more information.
if (major == 11 && minor <= 4) { GTEST_SKIP(); }
}

raft::random::uniform<ValueType>(handle, rng, v0.view(), 0, 1);
std::tuple<IndexType, ValueType, IndexType> stats;

Expand Down

0 comments on commit 0af5afb

Please sign in to comment.