From 392bd93e4d0e8e00343fa8b6398256bd73c8b7c9 Mon Sep 17 00:00:00 2001 From: Jake VanderPlas Date: Mon, 27 Mar 2023 10:15:43 -0700 Subject: [PATCH] [sparse] fix coo efficiency warning --- jax/experimental/sparse/coo.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/jax/experimental/sparse/coo.py b/jax/experimental/sparse/coo.py index 3f6e25855f48..2e9469213767 100644 --- a/jax/experimental/sparse/coo.py +++ b/jax/experimental/sparse/coo.py @@ -221,7 +221,7 @@ def _coo_todense_gpu_lowering(coo_todense_hlo, ctx, data, row, col, *, spinfo): shape = spinfo.shape[::-1] else: warnings.warn("coo_todense GPU lowering requires matrices with sorted rows or sorted cols. " - "To sort the rows in your matrix, use e.g. mat = mat._sort_rows(). Falling " + "To sort the rows in your matrix, use e.g. mat = mat._sort_indices(). Falling " "back to the default implementation.", CuSparseEfficiencyWarning) return _coo_todense_lowering(ctx, data, row, col, spinfo=spinfo) @@ -460,7 +460,7 @@ def _coo_matvec_gpu_lowering(coo_matvec_hlo, ctx, data, row, col, v, *, spinfo, shape = spinfo.shape[::-1] else: warnings.warn("coo_matvec GPU lowering requires matrices with sorted rows or sorted cols. " - "To sort the rows in your matrix, use e.g. mat = mat._sort_rows(). Falling " + "To sort the rows in your matrix, use e.g. mat = mat._sort_indices(). Falling " "back to the default implementation.", CuSparseEfficiencyWarning) return _coo_matvec_lowering(ctx, data, row, col, v, spinfo=spinfo, transpose=transpose) @@ -583,7 +583,7 @@ def _coo_matmat_gpu_lowering(coo_matmat_hlo, ctx, data, row, col, B, *, spinfo, shape = spinfo.shape[::-1] else: warnings.warn("coo_matmat GPU lowering requires matrices with sorted rows or sorted cols. " - "To sort the rows in your matrix, use e.g. mat = mat._sort_rows(). Falling " + "To sort the rows in your matrix, use e.g. mat = mat._sort_indices(). Falling " "back to the default implementation.", CuSparseEfficiencyWarning) return _coo_matmat_lowering(ctx, data, row, col, B, spinfo=spinfo, transpose=transpose)