Skip to content

Commit

Permalink
Skip gtests for Rmat Lanczos tests with cuda <= 11.4
Browse files Browse the repository at this point in the history
Similar to rapidsai#2520, also skip gtests for cuda 11.4 on the
RmatLanczosTest
  • Loading branch information
benfred committed Dec 9, 2024
1 parent ee45ce7 commit fcc3871
Showing 1 changed file with 12 additions and 0 deletions.
12 changes: 12 additions & 0 deletions cpp/test/sparse/solver/lanczos.cu
Original file line number Diff line number Diff line change
Expand Up @@ -109,6 +109,18 @@ class rmat_lanczos_tests

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(); }
}

uint64_t n_edges = sparsity * ((long long)(1 << r_scale) * (long long)(1 << c_scale));
uint64_t n_nodes = 1 << std::max(r_scale, c_scale);
uint64_t theta_len = std::max(r_scale, c_scale) * 4;
Expand Down

0 comments on commit fcc3871

Please sign in to comment.