diff --git a/README.md b/README.md index 6ea807535..36eb0e2ee 100644 --- a/README.md +++ b/README.md @@ -104,14 +104,17 @@ int main() { // 3. Run multiplication operation synchronously mgr.evalOpDefault( - { tensorInA, tensorInB, tensorOut }) + { tensorInA, tensorInB, tensorOut }); // 4. Map results back from GPU memory to print the results - mgr.evalOpDefault({ tensorInA, tensorInB, tensorOut }) + mgr.evalOpDefault({ tensorInA, tensorInB, tensorOut }); // Prints the output which is Output: { 2, 4, 6 } - std::cout << fmt::format("Output: {}", - tensorOut.data()) << std::endl; + std::cout<< "Output: { "; + for (const float& elem : tensorOut->data()) { + std::cout << elem << " "; + } + std::cout << "}" << std::endl; } ``` @@ -157,9 +160,9 @@ int main() { sq->begin(); // 5.2. Record batch commands - sq->recordrecordrecordrecord({ tensorInA }); + sq->record({ tensorInB }); + sq->record({ tensorOut }); // 5.3. Explicitly stop recording batch commands sq->end(); @@ -168,8 +171,11 @@ int main() { sq->eval(); // Prints the output which is Output: { 2, 4, 6 } - std::cout << fmt::format("Output: {}", - tensorOut.data()) << std::endl; + std::cout<< "Output: { "; + for (const float& elem : tensorOut->data()) { + std::cout << elem << " "; + } + std::cout << "}" << std::endl; } ```