From 885441b2c8689de2fe2f98ba3e5f86493ff62644 Mon Sep 17 00:00:00 2001 From: Jake VanderPlas Date: Tue, 29 Nov 2022 08:40:12 -0800 Subject: [PATCH] [sparse] ignore GPU warning in tests --- tests/sparse_test.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/sparse_test.py b/tests/sparse_test.py index 8e0b41da7..fa07268bf 100644 --- a/tests/sparse_test.py +++ b/tests/sparse_test.py @@ -2076,6 +2076,7 @@ class BCOOTest(sptu.SparseTestCase): rhs_dtype=all_dtypes, ) @jax.default_matmul_precision("float32") + @jtu.ignore_warning(category=sparse.CuSparseEfficiencyWarning) def test_bcoo_matmul(self, lhs_shape, lhs_dtype, rhs_shape, rhs_dtype): # TODO(b/259538729): Disable gpu test when type promotion is required. # BCOO type promotion calls `convert_element_type`, which further calls