Merge pull request #57 from GPUEngineering/hf/file-storage-mode

ruairimoran · web-flow · commit 3f408d6f8d0f · 2024-12-01T18:29:04.000Z
hf/ pass on requested storage mode
diff --git a/include/tensor.cuh b/include/tensor.cuh
@@ -644,7 +644,7 @@ template<typename T>
 DTensor<T> DTensor<T>::parseFromTextFile(std::string path_to_file,
                                          StorageMode mode) {
     auto parsedData = vectorFromFile<T>(path_to_file);
-    DTensor<T> tensorFromData(parsedData.data, parsedData.numRows, parsedData.numCols, parsedData.numMats);
+    DTensor<T> tensorFromData(parsedData.data, parsedData.numRows, parsedData.numCols, parsedData.numMats, mode);
     return tensorFromData;
 }
 

Original file line number	Diff line number	Diff line change
`@@ -644,7 +644,7 @@ template<typename T>`
`644`	`644`	`DTensor<T> DTensor<T>::parseFromTextFile(std::string path_to_file,`
`645`	`645`	`StorageMode mode) {`
`646`	`646`	`auto parsedData = vectorFromFile<T>(path_to_file);`
`647`		`- DTensor<T> tensorFromData(parsedData.data, parsedData.numRows, parsedData.numCols, parsedData.numMats);`
	`647`	`+ DTensor<T> tensorFromData(parsedData.data, parsedData.numRows, parsedData.numCols, parsedData.numMats, mode);`
`648`	`648`	`return tensorFromData;`
`649`	`649`	`}`
`650`	`650`