Skip to content

Commit b852d5d

Browse files
moved isMatrix flag
1 parent 201ec14 commit b852d5d

File tree

4 files changed

+7
-6
lines changed

4 files changed

+7
-6
lines changed

src/compiler/utils/CompilerUtils.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -164,8 +164,8 @@ template <> bool CompilerUtils::constantOrDefault<bool>(mlir::Value v, bool d) {
164164
// Other
165165
// **************************************************************************************************
166166

167-
[[maybe_unused]] FileMetaData CompilerUtils::getFileMetaData(mlir::Value filename) {
168-
return MetaDataParser::readMetaData(constantOrThrow<std::string>(filename));
167+
[[maybe_unused]] FileMetaData CompilerUtils::getFileMetaData(mlir::Value filename, bool isMatrix) {
168+
return MetaDataParser::readMetaData(constantOrThrow<std::string>(filename), ',', isMatrix);
169169
}
170170

171171
bool CompilerUtils::isMatrixComputation(mlir::Operation *v) {

src/compiler/utils/CompilerUtils.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ struct CompilerUtils {
116116
*/
117117
template <typename T> static T constantOrDefault(mlir::Value v, T d);
118118

119-
[[maybe_unused]] static FileMetaData getFileMetaData(mlir::Value filename);
119+
[[maybe_unused]] static FileMetaData getFileMetaData(mlir::Value filename, bool isMatrix = false);
120120

121121
/**
122122
* @brief Produces a string containing the C++ type name of the

src/parser/metadata/MetaDataParser.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,14 +22,14 @@
2222
#include <fstream>
2323
#include <iostream>
2424

25-
FileMetaData MetaDataParser::readMetaData(const std::string &filename_, char delim, size_t sampleRows) {
25+
FileMetaData MetaDataParser::readMetaData(const std::string &filename_, char delim, bool isMatrix, size_t sampleRows) {
2626
std::string metaFilename = filename_ + ".meta";
2727
std::ifstream ifs(metaFilename, std::ios::in);
2828
if (!ifs.good()) {
2929
int extv = extValue(&filename_[0]);
3030
// TODO: Support other file types than csv for metadata generation
3131
if (extv == 0) {
32-
FileMetaData fmd = generateFileMetaData(filename_, delim, sampleRows);
32+
FileMetaData fmd = generateFileMetaData(filename_, delim, sampleRows, isMatrix);
3333
try {
3434
writeMetaData(filename_, fmd);
3535
} catch (std::exception &e) {

src/parser/metadata/MetaDataParser.h

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,8 @@ class MetaDataParser {
6666
* @throws std::invalid_argument Thrown if the JSON file contains any
6767
* unexpected keys or if the file doesn't contain all the metadata.
6868
*/
69-
static FileMetaData readMetaData(const std::string &filename, char delim = ',', size_t sampleRows = std::numeric_limits<size_t>::max());
69+
static FileMetaData readMetaData(const std::string &filename, char delim = ',', bool isMatrix = false,
70+
size_t sampleRows = std::numeric_limits<size_t>::max());
7071
static FileMetaData readMetaDataFromString(const std::string &str);
7172
/**
7273
* @brief Saves the file meta data to the specified file.

0 commit comments

Comments
 (0)