diff --git a/cpp/test/CMakeLists.txt b/cpp/test/CMakeLists.txt index b93606dcc0..6f2d000ed0 100644 --- a/cpp/test/CMakeLists.txt +++ b/cpp/test/CMakeLists.txt @@ -122,10 +122,14 @@ if(BUILD_CUML_TESTS) ConfigureTest(PREFIX SG NAME GENETIC_PARAM_TEST PATH sg/genetic/param_test.cu OPTIONAL ML_INCLUDE) endif() - if(all_algo OR hdbscan_algo) - ConfigureTest(PREFIX SG NAME HDBSCAN_TEST PATH sg/hdbscan_test.cu OPTIONAL ML_INCLUDE) + if("${CMAKE_CUDA_COMPILER_VERSION}" VERSION_LESS_EQUAL "11.2") + # An HDBSCAN gtest is failing w/ CUDA 11.2 for some reason. + if(all_algo OR hdbscan_algo) + ConfigureTest(PREFIX SG NAME HDBSCAN_TEST PATH sg/hdbscan_test.cu OPTIONAL ML_INCLUDE) + endif() endif() + if(all_algo OR holtwinters_algo) ConfigureTest(PREFIX SG NAME HOLTWINTERS_TEST PATH sg/holtwinters_test.cu OPTIONAL ML_INCLUDE) endif() diff --git a/python/cuml/tests/test_hdbscan.py b/python/cuml/tests/test_hdbscan.py index c33401b1be..64dfaa905c 100644 --- a/python/cuml/tests/test_hdbscan.py +++ b/python/cuml/tests/test_hdbscan.py @@ -204,6 +204,8 @@ def test_hdbscan_blobs(nrows, ncols, nclusters, np.sort(cuml_agg.cluster_persistence_), rtol=0.01, atol=0.01) +@pytest.mark.skipif(cp.cuda.driver.get_build_version() <= 11020, + reason="Test failing on driver 11.2") @pytest.mark.parametrize('dataset', test_datasets.values()) @pytest.mark.parametrize('cluster_selection_epsilon', [0.0, 50.0, 150.0]) @pytest.mark.parametrize('min_samples_cluster_size_bounds', [(150, 150, 0), @@ -623,6 +625,8 @@ def test_all_points_membership_vectors_moons(nrows, sk_membership_vectors) +@pytest.mark.skipif(cp.cuda.driver.get_build_version() <= 11020, + reason="Test failing on driver 11.2") @pytest.mark.parametrize('nrows', [1000]) @pytest.mark.parametrize('n_points_to_predict', [200, 500]) @pytest.mark.parametrize('ncols', [10, 25]) diff --git a/python/cuml/tests/test_pca.py b/python/cuml/tests/test_pca.py index 033d2018c6..60c517f927 100644 --- a/python/cuml/tests/test_pca.py +++ b/python/cuml/tests/test_pca.py @@ -232,6 +232,8 @@ def test_pca_inverse_transform(datatype, input_type, 5e-5, with_sign=True) +@pytest.mark.skipif(cp.cuda.driver.get_build_version() <= 11020, + reason="Test failing on driver 11.2") @pytest.mark.parametrize('nrows', [4000, 8000]) @pytest.mark.parametrize('ncols', [5000, stress_param(20000)]) @pytest.mark.parametrize('whiten', [True, False])