Skip to content

Commit 7fc3d32

Browse files
authored
[mlir][sparse] update COO buffer reader doc (llvm#69664)
1 parent 1afb313 commit 7fc3d32

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

mlir/include/mlir/ExecutionEngine/SparseTensorRuntime.h

+5-4
Original file line numberDiff line numberDiff line change
@@ -120,15 +120,16 @@ MLIR_CRUNNERUTILS_EXPORT void _mlir_ciface_getSparseTensorReaderDimSizes(
120120
StridedMemRefType<index_type, 1> *out, void *p);
121121

122122
/// Reads the sparse tensor, stores the coordinates and values to the given
123-
/// memrefs. Returns a boolean to indicate whether the COO elements are sorted.
124-
#define DECL_GETNEXT(VNAME, V, CNAME, C) \
123+
/// memrefs of a COO in AoS format. Returns a boolean to indicate whether
124+
/// the COO elements are sorted.
125+
#define DECL_READTOBUFFERS(VNAME, V, CNAME, C) \
125126
MLIR_CRUNNERUTILS_EXPORT bool \
126127
_mlir_ciface_getSparseTensorReaderReadToBuffers##CNAME##VNAME( \
127128
void *p, StridedMemRefType<index_type, 1> *dim2lvlRef, \
128129
StridedMemRefType<index_type, 1> *lvl2dimRef, \
129130
StridedMemRefType<C, 1> *cref, StridedMemRefType<V, 1> *vref) \
130-
MLIR_SPARSETENSOR_FOREVERY_V_O(DECL_GETNEXT)
131-
#undef DECL_GETNEXT
131+
MLIR_SPARSETENSOR_FOREVERY_V_O(DECL_READTOBUFFERS)
132+
#undef DECL_READTOBUFFERS
132133

133134
/// Outputs the sparse tensor dim-rank, nse, and dim-shape.
134135
MLIR_CRUNNERUTILS_EXPORT void _mlir_ciface_outSparseTensorWriterMetaData(

0 commit comments

Comments
 (0)