@@ -449,9 +449,9 @@ void _mlir_ciface_outSparseTensorWriterMetaData(
449
449
ASSERT_NO_STRIDE (dimSizesRef);
450
450
assert (dimRank != 0 );
451
451
index_type *dimSizes = MEMREF_GET_PAYLOAD (dimSizesRef);
452
- SparseTensorWriter &file = *static_cast <SparseTensorWriter *>(p);
452
+ std::ostream &file = *static_cast <std::ostream *>(p);
453
453
file << dimRank << " " << nse << std::endl;
454
- for (index_type d = 0 ; d < dimRank - 1 ; ++d )
454
+ for (index_type d = 0 ; d < dimRank - 1 ; d++ )
455
455
file << dimSizes[d] << " " ;
456
456
file << dimSizes[dimRank - 1 ] << std::endl;
457
457
}
@@ -464,8 +464,8 @@ void _mlir_ciface_outSparseTensorWriterMetaData(
464
464
assert (p &&vref); \
465
465
ASSERT_NO_STRIDE (dimCoordsRef); \
466
466
const index_type *dimCoords = MEMREF_GET_PAYLOAD (dimCoordsRef); \
467
- SparseTensorWriter &file = *static_cast <SparseTensorWriter *>(p); \
468
- for (index_type d = 0 ; d < dimRank; ++d ) \
467
+ std::ostream &file = *static_cast <std::ostream *>(p); \
468
+ for (index_type d = 0 ; d < dimRank; d++ ) \
469
469
file << (dimCoords[d] + 1 ) << " " ; \
470
470
V *value = MEMREF_GET_PAYLOAD (vref); \
471
471
file << *value << std::endl; \
@@ -537,14 +537,14 @@ void delSparseTensorReader(void *p) {
537
537
}
538
538
539
539
void *createSparseTensorWriter (char *filename) {
540
- SparseTensorWriter *file =
540
+ std::ostream *file =
541
541
(filename[0 ] == 0 ) ? &std::cout : new std::ofstream (filename);
542
542
*file << " # extended FROSTT format\n " ;
543
543
return static_cast <void *>(file);
544
544
}
545
545
546
546
void delSparseTensorWriter (void *p) {
547
- SparseTensorWriter *file = static_cast <SparseTensorWriter *>(p);
547
+ std::ostream *file = static_cast <std::ostream *>(p);
548
548
file->flush ();
549
549
assert (file->good ());
550
550
if (file != &std::cout)
0 commit comments