[Mlir-commits] [mlir] 7fc3d32 - [mlir][sparse] update COO buffer reader doc (#69664)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Fri Oct 20 09:26:19 PDT 2023
Author: Aart Bik
Date: 2023-10-20T09:26:15-07:00
New Revision: 7fc3d32d531df9cd9afead78ad27df36e0e08242
URL: https://github.com/llvm/llvm-project/commit/7fc3d32d531df9cd9afead78ad27df36e0e08242
DIFF: https://github.com/llvm/llvm-project/commit/7fc3d32d531df9cd9afead78ad27df36e0e08242.diff
LOG: [mlir][sparse] update COO buffer reader doc (#69664)
Added:
Modified:
mlir/include/mlir/ExecutionEngine/SparseTensorRuntime.h
Removed:
################################################################################
diff --git a/mlir/include/mlir/ExecutionEngine/SparseTensorRuntime.h b/mlir/include/mlir/ExecutionEngine/SparseTensorRuntime.h
index 8955b79f091977b..5a620807f15ff7e 100644
--- a/mlir/include/mlir/ExecutionEngine/SparseTensorRuntime.h
+++ b/mlir/include/mlir/ExecutionEngine/SparseTensorRuntime.h
@@ -120,15 +120,16 @@ MLIR_CRUNNERUTILS_EXPORT void _mlir_ciface_getSparseTensorReaderDimSizes(
StridedMemRefType<index_type, 1> *out, void *p);
/// Reads the sparse tensor, stores the coordinates and values to the given
-/// memrefs. Returns a boolean to indicate whether the COO elements are sorted.
-#define DECL_GETNEXT(VNAME, V, CNAME, C) \
+/// memrefs of a COO in AoS format. Returns a boolean to indicate whether
+/// the COO elements are sorted.
+#define DECL_READTOBUFFERS(VNAME, V, CNAME, C) \
MLIR_CRUNNERUTILS_EXPORT bool \
_mlir_ciface_getSparseTensorReaderReadToBuffers##CNAME##VNAME( \
void *p, StridedMemRefType<index_type, 1> *dim2lvlRef, \
StridedMemRefType<index_type, 1> *lvl2dimRef, \
StridedMemRefType<C, 1> *cref, StridedMemRefType<V, 1> *vref) \
- MLIR_SPARSETENSOR_FOREVERY_V_O(DECL_GETNEXT)
-#undef DECL_GETNEXT
+ MLIR_SPARSETENSOR_FOREVERY_V_O(DECL_READTOBUFFERS)
+#undef DECL_READTOBUFFERS
/// Outputs the sparse tensor dim-rank, nse, and dim-shape.
MLIR_CRUNNERUTILS_EXPORT void _mlir_ciface_outSparseTensorWriterMetaData(
More information about the Mlir-commits
mailing list